Searched refs:iteration_shape (Results 1 – 2 of 2) sorted by relevance
324 Shape iteration_shape = ShapeUtil::MakeShape(keys_shape.element_type(), in EmitSortInPlace() local391 return gpu::ParallelLoopEmitter(compare_loop_body_emitter, iteration_shape, in EmitSortInPlace()
2939 Shape iteration_shape = keys_shape; in EmitSortFromMlir() local2943 iteration_shape.set_dimensions(dimension_to_sort, num_iterations_in_sort_dim); in EmitSortFromMlir()2944 uint64 num_iterations = ShapeUtil::ElementsIn(iteration_shape); in EmitSortFromMlir()