Searched refs:shape_index_a (Results 1 – 2 of 2) sorted by relevance
440 const ShapeIndex& shape_index_a,
510 const HloInstruction* hlo_a, const ShapeIndex& shape_index_a, in SharesSliceAtIndex() argument512 return GetUniqueSlice(hlo_a, shape_index_a).ConsumeValueOrDie() == in SharesSliceAtIndex()