Searched refs:launch_unrolled_kernel_for_multi_outputs (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Loops.cuh | 265 static inline void launch_unrolled_kernel_for_multi_outputs(int64_t N, const func_t& f, array_t dat… in launch_unrolled_kernel_for_multi_outputs() function
295 launch_unrolled_kernel_for_multi_outputs<num_outputs>(numel, f, data, input_calc, output_calc); in gpu_kernel_multiple_outputs_impl()
299 launch_unrolled_kernel_for_multi_outputs<num_outputs>(numel, f, data, input_calc, output_calc); in gpu_kernel_multiple_outputs_impl()
|