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