HomeSort by relevance Sort by last modified time
    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