Searched defs:sH (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/aten/src/ATen/native/ |
| D | GridSampler.h | 220 int64_t sH, in get_value_bounded()
238 int64_t sH, int64_t sW, int64_t H, int64_t W, in safe_add_2d()
247 int64_t sD, int64_t sH, int64_t sW, in safe_add_3d()
263 int64_t sH, in add_value_bounded()
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | GridSampler.cuh | 231 const scalar_t *data, scalar_t x, scalar_t y, int W, int H, int sW, int sH, in get_value_bounded()
250 int sH, int sW, int H, int W, in safe_add_2d()
266 int sD, int sH, int sW, int D, int H, int W, in safe_add_3d()
282 scalar_t* data, scalar_t x, scalar_t y, int W, int H, int sW, int sH, in add_value_bounded()
|
| /external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| D | Pooling.cpp | 54 int64_t sH, in spatial_dilated_max_pooling()
116 int64_t sH, in spatial_dilated_max_pooling3d()
191 int64_t sH, in q_maxpool_2d()
357 int64_t sH, in q_maxpool_3d()
|
| /external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/ |
| D | QuantizedOpKernels.cpp | 1461 int64_t sH, in _qmaxpool_2d_nhwc_kernel()
1561 int64_t sH, in qmaxpool_2d_nhwc_kernel()
1592 int64_t sH, in qmaxpool_3d_nthwc_kernel()
|