Searched refs:RequiredLeaves (Results 1 – 2 of 2) sorted by relevance
66 CHECK_EQ(flattened_list.size(), RequiredLeaves(tuple_shape)) in Tuple()68 << RequiredLeaves(tuple_shape); in Tuple()76 int64 leaf_count = RequiredLeaves(tuple_shape); in SingleTuple()201 int64 HloSharding::RequiredLeaves(const Shape& shape) { in RequiredLeaves() function in xla::HloSharding209 int64 shape_leaves = RequiredLeaves(shape); in CheckLeafCount()
258 static int64 RequiredLeaves(const Shape& shape);