OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:IsLeafIndex
(Results
1 - 4
of
4
) sorted by null
/external/tensorflow/tensorflow/compiler/xla/
shape_util_test.cc
414
TEST(ShapeUtilTest,
IsLeafIndex
) {
417
EXPECT_TRUE(ShapeUtil::
IsLeafIndex
(array_shape, {}));
421
EXPECT_FALSE(ShapeUtil::
IsLeafIndex
(tuple_shape, {}));
422
EXPECT_TRUE(ShapeUtil::
IsLeafIndex
(tuple_shape, {0}));
423
EXPECT_TRUE(ShapeUtil::
IsLeafIndex
(tuple_shape, {1}));
431
EXPECT_FALSE(ShapeUtil::
IsLeafIndex
(nested_tuple_shape, {}));
432
EXPECT_TRUE(ShapeUtil::
IsLeafIndex
(nested_tuple_shape, {0}));
433
EXPECT_FALSE(ShapeUtil::
IsLeafIndex
(nested_tuple_shape, {1}));
434
EXPECT_TRUE(ShapeUtil::
IsLeafIndex
(nested_tuple_shape, {1, 0}));
435
EXPECT_TRUE(ShapeUtil::
IsLeafIndex
(nested_tuple_shape, {1, 1}))
[
all
...]
shape_util.h
440
static bool
IsLeafIndex
(const Shape& shape, const ShapeIndex& index);
shape_util.cc
838
bool ShapeUtil::
IsLeafIndex
(const Shape& shape, const ShapeIndex& index) {
[
all
...]
/external/tensorflow/tensorflow/compiler/xla/service/
layout_assignment.cc
[
all
...]
Completed in 112 milliseconds