Searched defs:translate_idx (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/aten/src/ATen/native/ |
| D | SharedReduceOps.h | 143 static C10_DEVICE acc_t translate_idx(acc_t acc, int64_t /*base_idx*/) { in translate_idx() function
177 static C10_DEVICE acc_t translate_idx(acc_t acc, int64_t /*base_idx*/) { in translate_idx() function
210 static C10_DEVICE acc_t translate_idx(acc_t acc, int64_t /*base_idx*/) { in translate_idx() function
239 static C10_DEVICE acc_t translate_idx(acc_t acc, int64_t /*base_idx*/) { in translate_idx() function
270 static C10_DEVICE acc_t translate_idx(acc_t acc, int64_t /*base_idx*/) { in translate_idx() function
302 static C10_DEVICE acc_t translate_idx(acc_t acc, int64_t /*base_idx*/) { in translate_idx() function
332 static C10_DEVICE acc_t translate_idx(acc_t acc, int64_t /*base_idx*/) { in translate_idx() function
381 static C10_DEVICE acc_t translate_idx(acc_t acc, int64_t /*base_idx*/) { in translate_idx() function
406 static C10_DEVICE acc_t translate_idx(acc_t acc, int64_t /*base_idx*/) { in translate_idx() function
465 static C10_DEVICE arg_t translate_idx(arg_t a, int64_t base_idx) { in translate_idx() function
[all …]
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | Reduce.cuh | 263 static __device__ arg_t translate_idx(arg_t acc, int64_t /*idx*/) { in translate_idx() function
|