Searched refs:gather_slice_sizes_ (Results 1 – 2 of 2) sorted by relevance
1371 return gather_slice_sizes_; in gather_slice_sizes()1396 std::vector<int64> gather_slice_sizes_; variable
2318 absl::c_copy(slice_sizes, std::back_inserter(gather_slice_sizes_)); in HloGatherInstruction()