Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DReduce.cuh74 struct ReduceConfig { struct
75 static constexpr int BLOCK_X = 0;
76 static constexpr int BLOCK_Y = 1;
77 static constexpr int CTA = 2;
79 static constexpr int input_vec_size = 4;
81 ReduceConfig(int element_size_bytes, int num_outputs, int num_inputs) in ReduceConfig() function
85 int element_size_bytes;
86 int num_inputs;
87 int num_outputs;
88 int step_input = 1;
[all …]