Searched refs:memory_span (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | GridSampler.cuh | 253 const index_t memory_span) { in safe_add_2d() argument
257 memory_span, in safe_add_2d()
269 const index_t memory_span) { in safe_add_3d() argument
273 memory_span, in safe_add_3d()
287 const index_t memory_span) { in add_value_bounded() argument
295 safe_add_2d(data, iy, ix, sH, sW, H, W, delta, NC_offset, memory_span); in add_value_bounded()
|