Searched defs:stride_before (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | Indexing.cu | 52 int64_t numel, int64_t stride, int64_t stride_before, int64_t outer_dim, bool accumulate) { in indexing_backward_kernel()
130 int64_t numel, int64_t stride, int64_t stride_before, int64_t outer_dim, bool accumulate) { in indexing_backward_kernel_stride_1()
186 int64_t numel, int64_t stride, int64_t stride_before, int64_t outer_dim, bool accumulate) { in indexing_backward_kernel_small_stride()
230 int64_t numel, int64_t stride, int64_t stride_before, int64_t outer_dim, in indexing_backward_kernel_quantized()
|