Searched refs:IsLeafIndex (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 451 TEST(ShapeUtilTest, IsLeafIndex) { in TEST() argument
454 EXPECT_TRUE(ShapeUtil::IsLeafIndex(array_shape, {})); in TEST()
458 EXPECT_FALSE(ShapeUtil::IsLeafIndex(tuple_shape, {})); in TEST()
459 EXPECT_TRUE(ShapeUtil::IsLeafIndex(tuple_shape, {0})); in TEST()
460 EXPECT_TRUE(ShapeUtil::IsLeafIndex(tuple_shape, {1})); in TEST()
468 EXPECT_FALSE(ShapeUtil::IsLeafIndex(nested_tuple_shape, {})); in TEST()
469 EXPECT_TRUE(ShapeUtil::IsLeafIndex(nested_tuple_shape, {0})); in TEST()
470 EXPECT_FALSE(ShapeUtil::IsLeafIndex(nested_tuple_shape, {1})); in TEST()
471 EXPECT_TRUE(ShapeUtil::IsLeafIndex(nested_tuple_shape, {1, 0})); in TEST()
472 EXPECT_TRUE(ShapeUtil::IsLeafIndex(nested_tuple_shape, {1, 1})); in TEST()
|
| D | shape_util.cc | 818 bool ShapeUtil::IsLeafIndex(const Shape& shape, const ShapeIndex& index) { in IsLeafIndex() function in xla::ShapeUtil
837 if (IsLeafIndex(shape, index)) { in GetLeafShapes()
|
| D | shape_util.h | 499 static bool IsLeafIndex(const Shape& shape, const ShapeIndex& index);
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | bfloat16_propagation.cc | 419 if (!ShapeUtil::IsLeafIndex(parameter->shape(), index)) { in AdjustCalledComputationParameters()
459 if (!ShapeUtil::IsLeafIndex(hlo->shape(), index)) { in AdjustCalledComputationRoot()
|
| D | layout_assignment.cc | 895 if (ShapeUtil::IsLeafIndex(instruction->shape(), index)) { in CheckLayouts()
1205 if (ShapeUtil::IsLeafIndex(shape_layout.shape(), index)) { in PropagateUseConstraintToDefs()
|