Lines Matching refs:resultShape
61 SmallVectorImpl<int64_t> &resultShape) { in getBroadcastedShape() argument
70 resultShape.clear(); in getBroadcastedShape()
72 std::copy(shape1.begin(), shape1.end(), std::back_inserter(resultShape)); in getBroadcastedShape()
74 std::copy(shape2.begin(), shape2.end(), std::back_inserter(resultShape)); in getBroadcastedShape()
79 auto iR = resultShape.rbegin(); in getBroadcastedShape()
106 resultShape.clear(); in getBroadcastedShape()
175 SmallVector<int64_t, 4> resultShape; in getBroadcastedType() local
176 if (!getBroadcastedShape(getShape(type1), getShape(type2), resultShape)) in getBroadcastedType()
181 return VectorType::get(resultShape, elementType); in getBroadcastedType()
183 return RankedTensorType::get(resultShape, elementType); in getBroadcastedType()
252 SmallVector<int64_t, 4> resultShape; in verifyCompatibleOperandBroadcast() local
254 resultShape); in verifyCompatibleOperandBroadcast()
256 SmallVector<int64_t, 4> temp = resultShape; in verifyCompatibleOperandBroadcast()
257 if (!util::getBroadcastedShape(temp, getShape(other), resultShape)) in verifyCompatibleOperandBroadcast()
270 getShape(type).take_back(resultShape.size()); in verifyCompatibleOperandBroadcast()
271 if (!isCompatibleInferredReturnShape(resultShape, actualSuffix)) in verifyCompatibleOperandBroadcast()
275 << getShapeString(resultShape); in verifyCompatibleOperandBroadcast()