Searched refs:jitted_gpu_reduce_kernel (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | ReduceSumProdKernel.cu | 37 jitted_gpu_reduce_kernel<sum_name, scalar_t, scalar_t>( in operator ()()
70 jitted_gpu_reduce_kernel<nansum_name, scalar_t, scalar_t>( in operator ()()
94 jitted_gpu_reduce_kernel<prod_name, scalar_t, out_t>( in operator ()()
128 jitted_gpu_reduce_kernel<prod_name, scalar_t, scalar_t>(iter, func, 1.); in operator ()()
|
| H A D | Reduce.cuh | 1239 inline void jitted_gpu_reduce_kernel(TensorIterator& iter, const std::string& func, ident_t ident=0, in jitted_gpu_reduce_kernel() function
1292 jitted_gpu_reduce_kernel<name, scalar_t, out_scalar_t, vt0>(sub_iter, func, ident, in jitted_gpu_reduce_kernel()
|