OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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