Searched refs:standard_iteration_shape (Results 1 – 1 of 1) sorted by relevance
2927 Shape standard_iteration_shape = keys_shape; in EmitSortFromMlir() local2929 standard_iteration_shape.set_dimensions(dimension_to_sort, in EmitSortFromMlir()2932 standard_iteration_shape, ir_emitter_context_->gpu_device_info()); in EmitSortFromMlir()