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