Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dshape_util_test.cc349 TEST(ShapeUtilTest, SameDimensions) { in TEST() argument
350 EXPECT_TRUE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {}), in TEST()
352 EXPECT_TRUE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {}), in TEST()
354 EXPECT_TRUE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {1}), in TEST()
356 EXPECT_TRUE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {0}), in TEST()
358 EXPECT_TRUE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {2}), in TEST()
360 EXPECT_FALSE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {1}), in TEST()
362 EXPECT_FALSE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {0, 0}), in TEST()
364 EXPECT_FALSE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {1}), in TEST()
366 EXPECT_FALSE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {}), in TEST()
[all …]
Dshape_util.cc106 if (!ShapeUtil::SameDimensions(lhs, rhs)) { in CompareShapes()
610 /* static */ bool ShapeUtil::SameDimensions(const Shape& lhs, in SameDimensions() function in xla::ShapeUtil
620 return SameDimensions(lhs, rhs) && SameElementType(lhs, rhs); in Compatible()
630 return SameDimensions(lhs, rhs); in CompatibleIgnoringElementType()
Dshape_util.h211 static bool SameDimensions(const Shape& lhs, const Shape& rhs);
/external/tensorflow/tensorflow/compiler/xla/service/
Dreshape_mover.cc116 !ShapeUtil::SameDimensions(a->shape(), b->shape())) { in AreEquivalentReshapes()
123 return ShapeUtil::SameDimensions(a->operand(0)->shape(), in AreEquivalentReshapes()
152 if (!ShapeUtil::SameDimensions(operand->shape(), instruction->shape())) { in AllOperandsHaveEasyShapeChanges()
Dimplicit_broadcast_remover.cc75 if (!ShapeUtil::SameDimensions(hlo->shape(), operand->shape())) { in ReplaceImplicitBroadcastOperands()
Dhlo_evaluator.cc152 if (!ShapeUtil::SameDimensions(shape, operand->shape())) { in ElementWiseUnaryOpImpl()
342 TF_RET_CHECK(ShapeUtil::SameDimensions(operand->shape(), convert->shape())); in HandleConvert()
1971 if (!(ShapeUtil::SameDimensions(shape, rhs->shape()) && in ElementWiseBinaryOp()
1972 ShapeUtil::SameDimensions(lhs->shape(), rhs->shape()))) { in ElementWiseBinaryOp()
2006 if (!(ShapeUtil::SameDimensions(shape, lhs->shape()) && in ElementwiseTernaryOp()
2007 ShapeUtil::SameDimensions(lhs->shape(), rhs->shape()) && in ElementwiseTernaryOp()
2008 ShapeUtil::SameDimensions(rhs->shape(), ehs->shape()))) { in ElementwiseTernaryOp()
2314 if (!(ShapeUtil::SameDimensions(compare->shape(), rhs->shape()) && in HandleCompare()
2315 ShapeUtil::SameDimensions(lhs->shape(), rhs->shape()))) { in HandleCompare()
Duser_computation.cc3319 if (!ShapeUtil::SameDimensions(request.output_shape(), lhs->shape())) { in Visit()
3325 if (!ShapeUtil::SameDimensions(request.output_shape(), rhs->shape())) { in Visit()
3330 if (!ShapeUtil::SameDimensions(request.output_shape(), ehs->shape())) { in Visit()
3457 if (!ShapeUtil::SameDimensions(request.output_shape(), lhs->shape())) { in Visit()
3463 if (!ShapeUtil::SameDimensions(request.output_shape(), rhs->shape())) { in Visit()
Dalgebraic_simplifier.cc691 if (ShapeUtil::SameDimensions(hlo->shape(), dot->shape())) { in HandleDotStrengthReduction()
Dshape_inference.cc2403 if (ShapeUtil::SameDimensions(pred, on_true) || ShapeUtil::Rank(pred) == 0) { in InferSelectShape()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dloop_emitter.cc87 CHECK(ShapeUtil::SameDimensions(shape_, array.GetShape())); in LoopEmitter()