Home
last modified time | relevance | path

Searched refs:generate_reduction_code (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A Djit_utils.h110 std::string generate_reduction_code(
123 std::string generate_reduction_code(
H A Djit_utils.cpp1304 std::string generate_reduction_code( in generate_reduction_code() function
1314 return generate_reduction_code( in generate_reduction_code()
1329 std::string generate_reduction_code( in generate_reduction_code() function
H A DReduce.cuh925 auto code = at::cuda::jit::generate_reduction_code( in launch_jitted_reduce_kernel()