Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DReduce.cuh282 struct ReduceJitOp { struct
285 using InputCalculator = OffsetCalculator<1, uint32_t>;
286 using OutputCalculator = OffsetCalculator<2, uint32_t>;
288 using arg_t = at::opmath_type<scalar_t>;
290 static constexpr int input_vec_size = ReduceConfig::input_vec_size;
293 arg_t ident;
294 ReduceConfig config;
295 InputCalculator input_calc;
296 OutputCalculator output_calc;
297 const void* src;
[all …]