Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DMemoryAccess.cuh293 struct multi_outputs_unroll { struct
296 data_t data;
297 int remaining;
298 inp_calc_t input_offset_calculator;
299 out_calc_t output_offset_calculator;
300 LoadWithoutCast loader;
301 StoreWithoutCast storer;
303 __device__ multi_outputs_unroll(data_t data, int remaining, inp_calc_t ic, out_calc_t oc): in multi_outputs_unroll() argument
306 __device__ inline bool check_inbounds(int thread_work_elem) { in check_inbounds()
311 __device__ inline void load(args_t *args, int idx) { in load()
[all …]