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()
|