Searched refs:partitioned_slice_offsets (Results 1 – 1 of 1) sorted by relevance
2324 std::vector<int64> partitioned_slice_offsets; in HandleDynamicUpdateSlice() local2333 partitioned_slice_offsets.push_back( in HandleDynamicUpdateSlice()2387 if ((partitioned_slice_offsets[i] / per_partition_size) != in HandleDynamicUpdateSlice()2388 ((partitioned_slice_offsets[i] + update_shape.dimensions(dim) - 1) / in HandleDynamicUpdateSlice()