| /external/llvm-project/mlir/unittests/Dialect/ |
| 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()
|
| /external/llvm-project/mlir/lib/Dialect/ |
| D | Traits.cpp | 33 bool OpTrait::util::getBroadcastedShape(ArrayRef<int64_t> shape1, in getBroadcastedShape() function in OpTrait::util
150 if (!getBroadcastedShape(getShape(type1), getShape(type2), resultShape)) in getBroadcastedType()
211 (void)util::getBroadcastedShape(getShape(*rankedOperands.begin()), {}, in verifyCompatibleOperandBroadcast()
215 if (!util::getBroadcastedShape(temp, getShape(other), resultShape)) in verifyCompatibleOperandBroadcast()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | fold_broadcast.cc | 65 return RewriteOp(op, rewriter, OpTrait::util::getBroadcastedShape); in matchAndRewrite()
94 if (!OpTrait::util::getBroadcastedShape( in RewriteBatchMatMulV2Op()
128 return RewriteOp(op, rewriter, OpTrait::util::getBroadcastedShape); in RewriteEqOp()
|
| /external/llvm-project/mlir/include/mlir/Dialect/ |
| D | Traits.h | 47 bool getBroadcastedShape(ArrayRef<int64_t> shape1, ArrayRef<int64_t> shape2,
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | legalize_tf.cc | 607 if (!OpTrait::util::getBroadcastedShape(lhs_shape, rhs_shape, in matchAndRewrite()
676 if (!OpTrait::util::getBroadcastedShape(lhs_shape, rhs_shape, in matchAndRewrite()
682 if (!OpTrait::util::getBroadcastedShape(broadcasted_shape, cond_shape, in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_types.cc | 245 return OpTrait::util::getBroadcastedShape(lhs_rt.getShape(), in BroadcastCompatible()
|
| D | tf_ops_a_m.cc | 206 if (!OpTrait::util::getBroadcastedShape(x_batches, y_batches, in Verify()
661 if (!OpTrait::util::getBroadcastedShape(s0_shape, s1_shape, bcasted_shape)) in Verify()
696 OpTrait::util::getBroadcastedShape(s0_shape, s1_shape, bcasted_shape); in fold()
|
| D | tf_op_base.td | 605 if (!OpTrait::util::getBroadcastedShape(
|
| D | tf_ops_n_z.cc | 825 if (!OpTrait::util::getBroadcastedShape(cond_ranked_ty.getShape(), in InferSelectV2OpType()
|
| /external/llvm-project/mlir/lib/Dialect/Shape/IR/ |
| D | Shape.cpp | 333 if (!OpTrait::util::getBroadcastedShape(lhsShape, rhsShape, resultShape)) in fold()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 99 if (!OpTrait::util::getBroadcastedShape(current_shape, shape, in VerifyOperandsHaveSameShapesOrBroadcastableShape()
2401 !OpTrait::util::getBroadcastedShape(cond_shape, operand_shape, in BuildSelectV2Op()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 2581 if (!OpTrait::util::getBroadcastedShape(ranked_then_type.getShape(), in matchAndRewrite()
2587 if (!OpTrait::util::getBroadcastedShape(broadcast_then_else_shape, in matchAndRewrite()
2787 OpTrait::util::getBroadcastedShape(lhs_type.getShape().drop_back(2), in BroadcastBatchMatMulV2Operands()
|