Searched refs:sort_dim_elements (Results 1 – 1 of 1) sorted by relevance
2181 int64 sort_dim_elements = key_shape.dimensions(sort_dim); in HandleSort() local2182 increment[sort_dim] = sort_dim_elements; in HandleSort()2192 limit_indices[sort_dim] = sort_dim_elements; in HandleSort()2199 .Reshape({sort_dim_elements})); in HandleSort()2202 std::vector<int64> indices_to_sort(sort_dim_elements); in HandleSort()2250 slice_dimensions[sort_dim] = sort_dim_elements; in HandleSort()