Home
last modified time | relevance | path

Searched defs:arg_vec_t (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DReduce.cuh410 using arg_vec_t = at::detail::Array<arg_t, output_vec_size>; in run() typedef
569 using arg_vec_t = at::detail::Array<arg_t, output_vec_size>; in thread_reduce_impl() typedef
784 using arg_vec_t = at::detail::Array<arg_t, output_vec_size>; in global_reduce() typedef