Searched refs:IsLeafIndex (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 473 TEST(ShapeUtilTest, IsLeafIndex) { in TEST() argument
476 EXPECT_TRUE(ShapeUtil::IsLeafIndex(array_shape, {})); in TEST()
480 EXPECT_FALSE(ShapeUtil::IsLeafIndex(tuple_shape, {})); in TEST()
481 EXPECT_TRUE(ShapeUtil::IsLeafIndex(tuple_shape, {0})); in TEST()
482 EXPECT_TRUE(ShapeUtil::IsLeafIndex(tuple_shape, {1})); in TEST()
490 EXPECT_FALSE(ShapeUtil::IsLeafIndex(nested_tuple_shape, {})); in TEST()
491 EXPECT_TRUE(ShapeUtil::IsLeafIndex(nested_tuple_shape, {0})); in TEST()
492 EXPECT_FALSE(ShapeUtil::IsLeafIndex(nested_tuple_shape, {1})); in TEST()
493 EXPECT_TRUE(ShapeUtil::IsLeafIndex(nested_tuple_shape, {1, 0})); in TEST()
494 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 | 512 static bool IsLeafIndex(const Shape& shape, const ShapeIndex& index);
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_replication_analysis.cc | 290 if (!ShapeUtil::IsLeafIndex(param->shape(), index)) { in ComputeHloReplication()
302 if (!ShapeUtil::IsLeafIndex(param->shape(), index)) { in ComputeHloReplication()
|
| D | bfloat16_propagation.cc | 435 if (!ShapeUtil::IsLeafIndex(parameter->shape(), index)) { in AdjustCalledComputationParameters()
475 if (!ShapeUtil::IsLeafIndex(hlo->shape(), index)) { in AdjustCalledComputationRoot()
|
| D | layout_assignment.cc | 769 if (!ShapeUtil::IsLeafIndex(parameter_layout.shape(), shape_index) || in CheckParameterLayout()
947 if (ShapeUtil::IsLeafIndex(instruction->shape(), index)) { in CheckLayouts()
1262 if (ShapeUtil::IsLeafIndex(shape_layout.shape(), index)) { in PropagateUseConstraintToDefs()
2017 if (!ShapeUtil::IsLeafIndex(param_layout->shape(), shape_index)) { in PropagateComputationLayouts()
|
| D | bfloat16_normalization.cc | 110 if (ShapeUtil::IsLeafIndex(shape, index)) { in ShapeLeafCount()
|