Searched refs:IsLeafIndex (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 475 TEST(ShapeUtilTest, IsLeafIndex) { in TEST() argument
478 EXPECT_TRUE(ShapeUtil::IsLeafIndex(array_shape, {})); in TEST()
482 EXPECT_FALSE(ShapeUtil::IsLeafIndex(tuple_shape, {})); in TEST()
483 EXPECT_TRUE(ShapeUtil::IsLeafIndex(tuple_shape, {0})); in TEST()
484 EXPECT_TRUE(ShapeUtil::IsLeafIndex(tuple_shape, {1})); in TEST()
492 EXPECT_FALSE(ShapeUtil::IsLeafIndex(nested_tuple_shape, {})); in TEST()
493 EXPECT_TRUE(ShapeUtil::IsLeafIndex(nested_tuple_shape, {0})); in TEST()
494 EXPECT_FALSE(ShapeUtil::IsLeafIndex(nested_tuple_shape, {1})); in TEST()
495 EXPECT_TRUE(ShapeUtil::IsLeafIndex(nested_tuple_shape, {1, 0})); in TEST()
496 EXPECT_TRUE(ShapeUtil::IsLeafIndex(nested_tuple_shape, {1, 1})); in TEST()
|
| D | shape_util.h | 561 static bool IsLeafIndex(const Shape& shape, const ShapeIndex& index);
|
| D | shape_util.cc | 945 bool ShapeUtil::IsLeafIndex(const Shape& shape, const ShapeIndex& index) { in IsLeafIndex() function in xla::ShapeUtil
964 if (IsLeafIndex(shape, index)) { in GetLeafShapes()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_replication_analysis.cc | 340 if (!ShapeUtil::IsLeafIndex(param->shape(), index)) { in ComputeHloReplication()
352 if (!ShapeUtil::IsLeafIndex(param->shape(), index)) { in ComputeHloReplication()
|
| D | bfloat16_propagation.cc | 482 if (!ShapeUtil::IsLeafIndex(parameter->shape(), index)) { in AdjustCalledComputationParameters()
528 if (!ShapeUtil::IsLeafIndex(hlo->shape(), index)) { in AdjustCalledComputationRoot()
|
| D | layout_assignment.cc | 770 if (!ShapeUtil::IsLeafIndex(parameter_layout.shape(), shape_index) || in CheckParameterLayout()
948 if (ShapeUtil::IsLeafIndex(instruction->shape(), index)) { in CheckLayouts()
1263 if (ShapeUtil::IsLeafIndex(shape_layout.shape(), index)) { in PropagateUseConstraintToDefs()
2057 if (!ShapeUtil::IsLeafIndex(param_layout->shape(), shape_index)) { in PropagateComputationLayouts()
|
| D | bfloat16_normalization.cc | 110 if (ShapeUtil::IsLeafIndex(shape, index)) { in ShapeLeafCount()
|