Home
last modified time | relevance | path

Searched refs:jit_pwise_function (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DCUDAJitLoops.cuh102 fn_cache = at::cuda::jit::jit_pwise_function(code, desc.name); in launch_jitted_unrolled_kernel()
149 *fn_ptr = at::cuda::jit::jit_pwise_function(code, kernel_name); in launch_jitted_vectorized_kernel()
H A Djit_utils.h131 NvrtcFunction jit_pwise_function(
H A Djit_utils.cpp1470 NvrtcFunction jit_pwise_function( in jit_pwise_function() function
H A DReduce.cuh928 *fn_ptr = at::cuda::jit::jit_pwise_function(code, "reduction_" + desc.name); in launch_jitted_reduce_kernel()
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A Djiterator.cu67 *fn_ptr = at::cuda::jit::jit_pwise_function(code, kernel_name); in launch_jitted_vectorized_kernel_dynamic()
158 *fn_ptr = at::cuda::jit::jit_pwise_function(code, name); in launch_jitted_unrolled_kernel_dynamic()