| /external/tensorflow/tensorflow/compiler/xla/ |
| shape_util_test.cc | 349 TEST(ShapeUtilTest, SameDimensions) {
350 EXPECT_TRUE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {}),
352 EXPECT_TRUE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {}),
354 EXPECT_TRUE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {1}),
356 EXPECT_TRUE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {0}),
358 EXPECT_TRUE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {2}),
360 EXPECT_FALSE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {1}),
362 EXPECT_FALSE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {0, 0}),
364 EXPECT_FALSE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {1}),
366 EXPECT_FALSE(ShapeUtil::SameDimensions(ShapeUtil::MakeShape(S32, {})
[all...] |
| shape_util.cc | 106 if (!ShapeUtil::SameDimensions(lhs, rhs)) {
610 /* static */ bool ShapeUtil::SameDimensions(const Shape& lhs,
620 return SameDimensions(lhs, rhs) && SameElementType(lhs, rhs);
630 return SameDimensions(lhs, rhs);
[all...] |
| shape_util.h | 211 static bool SameDimensions(const Shape& lhs, const Shape& rhs);
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| implicit_broadcast_remover.cc | 75 if (!ShapeUtil::SameDimensions(hlo->shape(), operand->shape())) {
|
| reshape_mover.cc | 116 !ShapeUtil::SameDimensions(a->shape(), b->shape())) {
123 return ShapeUtil::SameDimensions(a->operand(0)->shape(),
152 if (!ShapeUtil::SameDimensions(operand->shape(), instruction->shape())) {
|
| hlo_evaluator.cc | 152 if (!ShapeUtil::SameDimensions(shape, operand->shape())) {
342 TF_RET_CHECK(ShapeUtil::SameDimensions(operand->shape(), convert->shape()));
[all...] |
| user_computation.cc | [all...] |
| algebraic_simplifier.cc | 691 if (ShapeUtil::SameDimensions(hlo->shape(), dot->shape())) {
[all...] |
| shape_inference.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| loop_emitter.cc | 87 CHECK(ShapeUtil::SameDimensions(shape_, array.GetShape()));
|
