Searched refs:launch_jitted_pwise_function (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | CUDAJitLoops.cuh | 107 at::cuda::jit::launch_jitted_pwise_function(fn_cache, args.data(), {grid, 1u, 1u}, in launch_jitted_unrolled_kernel()
155 at::cuda::jit::launch_jitted_pwise_function( in launch_jitted_vectorized_kernel()
169 at::cuda::jit::launch_jitted_pwise_function( in launch_jitted_vectorized_kernel()
|
| H A D | jit_utils.h | 135 void launch_jitted_pwise_function(
|
| H A D | jit_utils.cpp | 1637 void launch_jitted_pwise_function( in launch_jitted_pwise_function() function
|
| H A D | Reduce.cuh | 933 at::cuda::jit::launch_jitted_pwise_function(*fn_ptr, args, grid, block, shared_memory); in launch_jitted_reduce_kernel()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/ |
| H A D | jiterator.cu | 88 …at::cuda::jit::launch_jitted_pwise_function(*fn_ptr, args.get(), {grid, 1u, 1u}, {num_threads(), 1… in launch_jitted_vectorized_kernel_dynamic()
114 …at::cuda::jit::launch_jitted_pwise_function(*fn_ptr, args.get(), {grid, 1u, 1u}, {num_threads(), 1… in launch_jitted_vectorized_kernel_dynamic()
181 …at::cuda::jit::launch_jitted_pwise_function(*fn_ptr, args.get(), {grid, 1u, 1u}, {num_threads(), 1… in launch_jitted_unrolled_kernel_dynamic()
|