| /external/tensorflow/tensorflow/compiler/xla/ |
| layout_util_test.cc | 56 EXPECT_TRUE(LayoutUtil::LayoutsInShapesEqual(tuple0, tuple0));
57 EXPECT_FALSE(LayoutUtil::LayoutsInShapesEqual(tuple0, tuple1));
58 EXPECT_FALSE(LayoutUtil::LayoutsInShapesEqual(tuple0, tuple2));
59 EXPECT_FALSE(LayoutUtil::LayoutsInShapesEqual(tuple1, tuple0));
60 EXPECT_FALSE(LayoutUtil::LayoutsInShapesEqual(tuple2, tuple0));
62 EXPECT_TRUE(LayoutUtil::LayoutsInShapesEqual(tuple1, tuple1));
63 EXPECT_FALSE(LayoutUtil::LayoutsInShapesEqual(tuple1, tuple2));
64 EXPECT_FALSE(LayoutUtil::LayoutsInShapesEqual(tuple2, tuple1));
67 EXPECT_TRUE(LayoutUtil::LayoutsInShapesEqual(tuple2, tuple2));
68 EXPECT_TRUE(LayoutUtil::LayoutsInShapesEqual(tuple2, other_tuple2))
[all...] |
| layout_util.h | 187 static bool LayoutsInShapesEqual(const Shape& lhs, const Shape& rhs);
|
| layout_util.cc | 413 /* static */ bool LayoutUtil::LayoutsInShapesEqual(const Shape& lhs,
421 if (!LayoutsInShapesEqual(lhs.tuple_shapes(i), rhs.tuple_shapes(i))) {
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| cudnn_batchnorm_thunk.cc | 97 LayoutUtil::LayoutsInShapesEqual(hlo->shape(), hlo->operand(0)->shape()));
156 CHECK(LayoutUtil::LayoutsInShapesEqual(hlo->shape().tuple_shapes(0),
238 CHECK(LayoutUtil::LayoutsInShapesEqual(hlo->shape().tuple_shapes(0),
240 CHECK(LayoutUtil::LayoutsInShapesEqual(hlo->shape().tuple_shapes(0),
|
| ir_emitter_unnested.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| layout_assignment_test.cc | 240 LayoutUtil::LayoutsInShapesEqual(constant0->shape(), constant1->shape()));
243 EXPECT_TRUE(LayoutUtil::LayoutsInShapesEqual(
245 EXPECT_TRUE(LayoutUtil::LayoutsInShapesEqual(
281 EXPECT_TRUE(LayoutUtil::LayoutsInShapesEqual(result_shape, select->shape()));
[all...] |
| layout_assignment.cc | 677 TF_RET_CHECK(LayoutUtil::LayoutsInShapesEqual(
703 LayoutUtil::LayoutsInShapesEqual(init_shape, while_inst->shape()));
733 TF_RET_CHECK(LayoutUtil::LayoutsInShapesEqual(
736 TF_RET_CHECK(LayoutUtil::LayoutsInShapesEqual(
[all...] |
| service.cc | [all...] |
| hlo_evaluator.cc | [all...] |
| hlo_evaluator_test.cc | 525 ASSERT_TRUE(LayoutUtil::LayoutsInShapesEqual(input_literal.shape(),
545 ASSERT_FALSE(LayoutUtil::LayoutsInShapesEqual(input_literal.shape(),
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| ir_emitter.cc | 537 LayoutUtil::LayoutsInShapesEqual(keys_shape, operand->shape()));
538 TF_RET_CHECK(LayoutUtil::LayoutsInShapesEqual(
[all...] |
