HomeSort by relevance Sort by last modified time
    Searched refs:LayoutsInShapesEqual (Results 1 - 9 of 9) sorted by null

  /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 191 static bool LayoutsInShapesEqual(const Shape& lhs, const Shape& rhs);
layout_util.cc 424 /* static */ bool LayoutUtil::LayoutsInShapesEqual(const Shape& lhs,
435 if (!LayoutsInShapesEqual(lhs.tuple_shapes(i), rhs.tuple_shapes(i))) {
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_batchnorm_thunk.cc 98 LayoutUtil::LayoutsInShapesEqual(hlo->shape(), hlo->operand(0)->shape()));
154 CHECK(LayoutUtil::LayoutsInShapesEqual(hlo->shape().tuple_shapes(0),
234 CHECK(LayoutUtil::LayoutsInShapesEqual(hlo->shape().tuple_shapes(0),
236 CHECK(LayoutUtil::LayoutsInShapesEqual(hlo->shape().tuple_shapes(0),
  /external/tensorflow/tensorflow/compiler/xla/service/
layout_assignment_test.cc 211 LayoutUtil::LayoutsInShapesEqual(constant0->shape(), constant1->shape()));
214 EXPECT_TRUE(LayoutUtil::LayoutsInShapesEqual(
216 EXPECT_TRUE(LayoutUtil::LayoutsInShapesEqual(
252 EXPECT_TRUE(LayoutUtil::LayoutsInShapesEqual(result_shape, select->shape()));
    [all...]
layout_assignment.cc 704 LayoutUtil::LayoutsInShapesEqual(init_shape, while_inst->shape()));
741 TF_RET_CHECK(LayoutUtil::LayoutsInShapesEqual(
744 TF_RET_CHECK(LayoutUtil::LayoutsInShapesEqual(
    [all...]
hlo_evaluator_test.cc 424 ASSERT_TRUE(LayoutUtil::LayoutsInShapesEqual(input_literal->shape(),
444 ASSERT_FALSE(LayoutUtil::LayoutsInShapesEqual(input_literal->shape(),
    [all...]
service.cc     [all...]
hlo_evaluator.cc 347 if (LayoutUtil::LayoutsInShapesEqual(result->shape(), convert->shape())) {
    [all...]

Completed in 349 milliseconds