Searched defs:oheight_for_shape_check (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/aten/src/ATen/native/ |
| D | AveragePool3d.cpp | 131 …const int64_t oheight_for_shape_check = pooling_output_shape<int64_t>(iheight, kH, padH, dH, 1, ce… in TORCH_META_FUNC() local
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | AveragePool3d.cu | 505 …const int64_t oheight_for_shape_check = pooling_output_shape<int64_t>(iheight, kH, padH, dH, 1, ce… in TORCH_IMPL_FUNC() local
|