Searched refs:index_shapes (Results 1 – 3 of 3) sorted by relevance
850 Cast<HloDynamicSliceInstruction>(dynamic_slice)->index_shapes(), in HandleDynamicSlice()862 ->index_shapes())); in HandleDynamicUpdateSlice()
1608 Cast<HloDynamicSliceInstruction>(dynamic_slice)->index_shapes(),1672 ->index_shapes()));
1588 std::vector<Shape> index_shapes() const { in index_shapes() function