Home
last modified time | relevance | path

Searched refs:SameDimensions (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dgather_expander_test.cc95 EXPECT_TRUE(ShapeUtil::SameDimensions( in TEST()
99 EXPECT_TRUE(ShapeUtil::SameDimensions( in TEST()
103 EXPECT_TRUE(ShapeUtil::SameDimensions( in TEST()
Dreshape_mover.cc119 !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()
Dhlo_evaluator.h314 TF_RET_CHECK(ShapeUtil::SameDimensions(shape, operand->shape())); in ElementWiseUnaryOpImpl()
Dhlo_evaluator_typed_visitor.h248 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()
Dhlo_evaluator.cc676 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()
Dshape_inference.cc1050 if (!ShapeUtil::SameDimensions(*operand_shapes[0], in InferVariadicOpShape()
2060 if (!ShapeUtil::SameDimensions(*reduced_args[0], *reduced_args[i])) { in InferReduceShape()
Dhlo_verifier.cc379 if (!ShapeUtil::SameDimensions(sort->operand(0)->shape(), in HandleSort()
Dalgebraic_simplifier.cc1179 if (!ShapeUtil::SameDimensions(hlo->shape(), dot->shape())) { in HandleDotStrengthReduction()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util_test.cc386 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 …]
Dshape.cc164 if (!ShapeUtil::SameDimensions(lhs, rhs)) { in operator ()()
Dshape_util.h247 static bool SameDimensions(const Shape& lhs, const Shape& rhs);
Dshape_util.cc490 /* static */ bool ShapeUtil::SameDimensions(const Shape& lhs, in SameDimensions() function in xla::ShapeUtil
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dloop_emitter.cc81 CHECK(ShapeUtil::SameDimensions(shape_, array.GetShape())) in LoopEmitter()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc539 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/
Dir_emitter_unnested.cc3575 return ShapeUtil::SameDimensions(op_shape, parameter_shape) && in NumInputsInvolveInOnlyElementwiseOps()