| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | gather_expander_test.cc | 104 EXPECT_TRUE(ShapeUtil::SameDimensions( in TEST_F()
108 EXPECT_TRUE(ShapeUtil::SameDimensions( in TEST_F()
112 EXPECT_TRUE(ShapeUtil::SameDimensions( in TEST_F()
|
| D | reshape_mover.cc | 120 !ShapeUtil::SameDimensions(a->shape(), b->shape())) { in AreEquivalentReshapes()
127 return ShapeUtil::SameDimensions(a->operand(0)->shape(), in AreEquivalentReshapes()
286 if (!ShapeUtil::SameDimensions(operand->shape(), instruction->shape())) { in IsReshapeMoveCandidate()
|
| D | hlo_evaluator_typed_visitor.h | 307 TF_RET_CHECK(ShapeUtil::SameDimensions(operand->shape(), convert->shape()));
2436 TF_RET_CHECK(ShapeUtil::SameDimensions(shape, rhs->shape()));
2437 TF_RET_CHECK(ShapeUtil::SameDimensions(lhs->shape(), rhs->shape()));
2461 TF_RET_CHECK(ShapeUtil::SameDimensions(shape, lhs->shape()));
2462 TF_RET_CHECK(ShapeUtil::SameDimensions(lhs->shape(), rhs->shape()));
2463 TF_RET_CHECK(ShapeUtil::SameDimensions(rhs->shape(), ehs->shape()));
|
| D | hlo_evaluator.h | 439 TF_RET_CHECK(ShapeUtil::SameDimensions(shape, operand->shape())); in ElementWiseUnaryOpImpl()
|
| D | shape_inference.cc | 1096 if (!ShapeUtil::SameDimensions(*operand_shapes[0], in InferVariadicOpShape()
2233 if (!ShapeUtil::SameDimensions(*reduced_args[0], *reduced_args[i])) { in InferReduceShape()
2310 if (!ShapeUtil::SameDimensions(*operands[0], *operands[i])) { in InferReduceWindowShape()
|
| D | hlo_evaluator.cc | 1399 DCHECK(ShapeUtil::SameDimensions(compare->shape(), rhs->shape()) && in HandleCompare()
1400 ShapeUtil::SameDimensions(lhs->shape(), rhs->shape())); in HandleCompare()
3460 TF_RET_CHECK(ShapeUtil::SameDimensions(sort->operand(0)->shape(), in HandleSort()
|
| D | hlo_verifier.cc | 938 if (!ShapeUtil::SameDimensions(sort->operand(0)->shape(), in HandleSort()
|
| D | algebraic_simplifier.cc | 919 ShapeUtil::SameDimensions(lhs_update_window, rhs_update_window)) { in HandleAdd()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 425 TEST(ShapeUtilTest, SameDimensions) { in TEST() argument
426 EXPECT_TRUE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {}), in TEST()
428 EXPECT_TRUE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {}), in TEST()
430 EXPECT_TRUE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {1}), in TEST()
432 EXPECT_TRUE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {0}), in TEST()
434 EXPECT_TRUE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {2}), in TEST()
436 EXPECT_FALSE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {1}), in TEST()
438 EXPECT_FALSE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {0, 0}), in TEST()
440 EXPECT_FALSE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {1}), in TEST()
442 EXPECT_FALSE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {}), in TEST()
[all …]
|
| D | shape.cc | 169 if (!ShapeUtil::SameDimensions(lhs, rhs)) { in operator ()()
|
| D | shape_util.h | 158 static bool SameDimensions(const Shape& lhs, const Shape& rhs);
|
| D | shape_util.cc | 722 /* static */ bool ShapeUtil::SameDimensions(const Shape& lhs, in SameDimensions() function in xla::ShapeUtil
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | loop_emitter.cc | 62 CHECK(ShapeUtil::SameDimensions(shape_, array.GetShape())) in LoopEmitter()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 736 if (!ShapeUtil::SameDimensions(shape, *updated_lhs_shape)) { in BinaryOp()
742 if (!ShapeUtil::SameDimensions(shape, *updated_rhs_shape)) { in BinaryOp()
2270 if (!ShapeUtil::SameDimensions(output_shape, *shape)) { in Map()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 4370 return ShapeUtil::SameDimensions(op_shape, parameter_shape) && in NumInputsInvolveInOnlyElementwiseOps()
|