/external/tensorflow/tensorflow/compiler/xla/service/ |
D | gather_expander_test.cc | 95 EXPECT_TRUE(ShapeUtil::SameDimensions( in TEST() 99 EXPECT_TRUE(ShapeUtil::SameDimensions( in TEST() 103 EXPECT_TRUE(ShapeUtil::SameDimensions( in TEST()
|
D | reshape_mover.cc | 119 !ShapeUtil::SameDimensions(a->shape(), b->shape())) { in AreEquivalentReshapes() 126 return ShapeUtil::SameDimensions(a->operand(0)->shape(), in AreEquivalentReshapes() 287 if (!ShapeUtil::SameDimensions(operand->shape(), instruction->shape())) { in IsReshapeMoveCandidate()
|
D | hlo_evaluator.h | 314 TF_RET_CHECK(ShapeUtil::SameDimensions(shape, operand->shape())); in ElementWiseUnaryOpImpl()
|
D | hlo_evaluator_typed_visitor.h | 248 TF_RET_CHECK(ShapeUtil::SameDimensions(operand->shape(), convert->shape())); in HandleConvert() 258 TF_RET_CHECK(ShapeUtil::SameDimensions(operand->shape(), convert->shape())); in HandleBitcastConvert() 2950 TF_RET_CHECK(ShapeUtil::SameDimensions(shape, rhs->shape())); in ElementWiseBinaryOp() 2951 TF_RET_CHECK(ShapeUtil::SameDimensions(lhs->shape(), rhs->shape())); in ElementWiseBinaryOp() 2975 TF_RET_CHECK(ShapeUtil::SameDimensions(shape, lhs->shape())); in ElementwiseTernaryOp() 2976 TF_RET_CHECK(ShapeUtil::SameDimensions(lhs->shape(), rhs->shape())); in ElementwiseTernaryOp() 2977 TF_RET_CHECK(ShapeUtil::SameDimensions(rhs->shape(), ehs->shape())); in ElementwiseTernaryOp()
|
D | hlo_evaluator.cc | 676 DCHECK(ShapeUtil::SameDimensions(compare->shape(), rhs->shape()) && in HandleCompare() 677 ShapeUtil::SameDimensions(lhs->shape(), rhs->shape())); in HandleCompare() 1440 TF_RET_CHECK(ShapeUtil::SameDimensions(sort->operand(0)->shape(), in HandleSort()
|
D | shape_inference.cc | 1050 if (!ShapeUtil::SameDimensions(*operand_shapes[0], in InferVariadicOpShape() 2060 if (!ShapeUtil::SameDimensions(*reduced_args[0], *reduced_args[i])) { in InferReduceShape()
|
D | hlo_verifier.cc | 379 if (!ShapeUtil::SameDimensions(sort->operand(0)->shape(), in HandleSort()
|
D | algebraic_simplifier.cc | 1179 if (!ShapeUtil::SameDimensions(hlo->shape(), dot->shape())) { in HandleDotStrengthReduction()
|
/external/tensorflow/tensorflow/compiler/xla/ |
D | shape_util_test.cc | 386 TEST(ShapeUtilTest, SameDimensions) { in TEST() argument 387 EXPECT_TRUE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {}), in TEST() 389 EXPECT_TRUE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {}), in TEST() 391 EXPECT_TRUE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {1}), in TEST() 393 EXPECT_TRUE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {0}), in TEST() 395 EXPECT_TRUE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {2}), in TEST() 397 EXPECT_FALSE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {1}), in TEST() 399 EXPECT_FALSE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {0, 0}), in TEST() 401 EXPECT_FALSE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {1}), in TEST() 403 EXPECT_FALSE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {}), in TEST() [all …]
|
D | shape.cc | 164 if (!ShapeUtil::SameDimensions(lhs, rhs)) { in operator ()()
|
D | shape_util.h | 247 static bool SameDimensions(const Shape& lhs, const Shape& rhs);
|
D | shape_util.cc | 490 /* 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 | 81 CHECK(ShapeUtil::SameDimensions(shape_, array.GetShape())) in LoopEmitter()
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | xla_builder.cc | 539 if (!ShapeUtil::SameDimensions(shape, updated_lhs_shape)) { in BinaryOp() 544 if (!ShapeUtil::SameDimensions(shape, updated_rhs_shape)) { in BinaryOp() 569 !ShapeUtil::SameDimensions(shape, lhs_shape)) { in TernaryOp() 574 !ShapeUtil::SameDimensions(shape, rhs_shape)) { in TernaryOp() 579 !ShapeUtil::SameDimensions(shape, ehs_shape)) { in TernaryOp() 1770 if (!ShapeUtil::SameDimensions(output_shape, shape)) { in Map()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | ir_emitter_unnested.cc | 3575 return ShapeUtil::SameDimensions(op_shape, parameter_shape) && in NumInputsInvolveInOnlyElementwiseOps()
|