Searched refs:buffer_rank (Results 1 – 1 of 1) sorted by relevance
144 int64 buffer_rank = buffer_shape.rank(); in XlaScatter() local145 int64 num_window_dims_in_updates = buffer_rank - num_index_dims; in XlaScatter()152 for (int64 dim = num_index_dims; dim < buffer_rank; ++dim) { in XlaScatter()