Searched refs:num_iterations_in_sort_dim (Results 1 – 3 of 3) sorted by relevance
41 int64 num_iterations_in_sort_dim, int64 tile_size,
298 int64 num_iterations_in_sort_dim, const int64 tile_size, in EmitSortInPlace() argument321 dimensions_in_iteration_order[dim] = num_iterations_in_sort_dim; in EmitSortInPlace()
2942 uint64 num_iterations_in_sort_dim = rounded_bound / 2; in EmitSortFromMlir() local2943 iteration_shape.set_dimensions(dimension_to_sort, num_iterations_in_sort_dim); in EmitSortFromMlir()3010 xor_masks.size() > 1 ? num_iterations_in_sort_dim in EmitSortFromMlir()