Searched defs:istrideH (Results 1 – 8 of 8) sorted by relevance
| /external/pytorch/aten/src/ATen/native/ |
| D | AdaptiveMaxPooling3d.cpp | 97 int64_t istrideH, in adaptive_max_pool3d_single_out_frame()
176 int64_t istrideH, in adaptive_max_pool3d_out_frame()
272 int64_t istrideH = 0; in TORCH_IMPL_FUNC() local
|
| D | AdaptiveAveragePooling3d.cpp | 39 int64_t istrideH, in adaptive_avg_pool3d_out_frame()
119 int64_t istrideH = input.stride(-2); in adaptive_avg_pool3d_out_cpu_template() local
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | AdaptiveMaxPooling2d.cu | 49 int64_t istrideD, int64_t istrideH, int64_t istrideW) in adaptivemaxpool()
231 int64_t istrideH = input.stride(1); in TORCH_IMPL_FUNC() local
276 int64_t istrideH = input_.stride(2); in TORCH_IMPL_FUNC() local
|
| D | AdaptiveMaxPooling3d.cu | 51 int64_t istrideT, int64_t istrideH, int64_t istrideW, in adaptivemaxpool()
131 int64_t istrideT, int64_t istrideH, int64_t istrideW) in adaptivemaxpool_loop()
325 int64_t istrideD, istrideT, istrideH, istrideW; in TORCH_IMPL_FUNC() local
|
| D | AdaptiveAveragePooling3d.cu | 57 int64_t istrideT, int64_t istrideH, int64_t istrideW, in adaptiveaveragepool()
124 int64_t istrideD, int64_t istrideT, int64_t istrideH, int64_t istrideW) { in adaptiveaveragepool_loop()
367 int64_t istrideD, istrideT, istrideH, istrideW; in adaptive_avg_pool3d_out_cuda_template() local
|
| D | AdaptiveAveragePooling.cu | 56 int64_t istrideD, int64_t istrideH, int64_t istrideW) in adaptive_average_pool()
475 int64_t istrideH = input_.stride(2); in adaptive_avg_pool2d_out_cuda_template() local
560 int64_t istrideH = input_.stride(-2); in adaptive_avg_pool2d_out_cuda_template() local
|
| /external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| D | AdaptiveAveragePooling.cpp | 71 int64_t istrideH, in adaptive_avg_pool_single_out_frame()
|
| /external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/ |
| D | QuantizedOpKernels.cpp | 1827 int64_t istrideH,
1934 int64_t istrideH,
1971 int64_t istrideH,
2021 int istrideH = strideW * inputWidth; local
|