Searched refs:sort_dim_elements (Results 1 – 1 of 1) sorted by relevance
1461 int64 sort_dim_elements = key_shape.dimensions(sort_dim); in HandleSort() local1462 increment[sort_dim] = sort_dim_elements; in HandleSort()1472 limit_indices[sort_dim] = sort_dim_elements; in HandleSort()1479 .Reshape({sort_dim_elements})); in HandleSort()1482 std::vector<int64> indices_to_sort(sort_dim_elements); in HandleSort()1530 slice_dimensions[sort_dim] = sort_dim_elements; in HandleSort()