Searched refs:getBroadcastedShape (Results 1 – 4 of 4) sorted by relevance
| /llvm-project-15.0.7/mlir/unittests/Dialect/ |
| H A D | BroadcastShapeTest.cpp | 20 ASSERT_TRUE(getBroadcastedShape({}, {}, result)); in TEST()
26 ASSERT_TRUE(getBroadcastedShape({}, {4}, result)); in TEST()
32 ASSERT_TRUE(getBroadcastedShape({}, {3, 5, 4}, result)); in TEST()
38 ASSERT_TRUE(getBroadcastedShape({1, 7, 8, 9}, {8, 9}, result)); in TEST()
44 ASSERT_TRUE(getBroadcastedShape({8, 1, 2, 1, 4}, {5, 1, 7, 1}, result)); in TEST()
51 getBroadcastedShape({1, 2, -1, -1, -1}, {-1, -1, -1, 4, 1}, result)); in TEST()
57 ASSERT_FALSE(getBroadcastedShape({4, 3, 5, 5}, {3, 5, 4}, result)); in TEST()
63 ASSERT_FALSE(getBroadcastedShape({4, 3, 5, 5}, {3, 7, 5}, result)); in TEST()
69 ASSERT_FALSE(getBroadcastedShape({3, 5, 5}, {4, 5, 5}, result)); in TEST()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/ |
| H A D | Traits.cpp | 59 bool OpTrait::util::getBroadcastedShape(ArrayRef<int64_t> shape1, in getBroadcastedShape() function in OpTrait::util
176 if (!getBroadcastedShape(getShape(type1), getShape(type2), resultShape)) in getBroadcastedType()
253 (void)util::getBroadcastedShape(getShape(*rankedOperands.begin()), {}, in verifyCompatibleOperandBroadcast()
257 if (!util::getBroadcastedShape(temp, getShape(other), resultShape)) in verifyCompatibleOperandBroadcast()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/ |
| H A D | Traits.h | 47 bool getBroadcastedShape(ArrayRef<int64_t> shape1, ArrayRef<int64_t> shape2,
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Shape/IR/ |
| H A D | Shape.cpp | 663 if (!OpTrait::util::getBroadcastedShape(lhsShape, rhsShape, resultShape)) in fold()
746 if (OpTrait::util::getBroadcastedShape( in matchAndRewrite()
|