Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DReduce.cuh488 return {input_vectorized_thread_reduce_impl(data)}; in thread_reduce()
502 C10_DEVICE arg_t input_vectorized_thread_reduce_impl(const scalar_t* data) const { in input_vectorized_thread_reduce_impl() function