Searched defs:poolSizeH (Results 1 – 4 of 4) sorted by relevance
38 int64_t poolSizeH = pool_size[0]; in TORCH_META_FUNC() local138 int poolSizeW, int poolSizeH) { in fractional_max_pool2d_out_single_batch_frame()199 int poolSizeW, int poolSizeH) { in fractional_max_pool2d_out_frame()298 int64_t poolSizeH = pool_size[0]; in TORCH_IMPL_FUNC() local
37 int64_t poolSizeH = pool_size[1]; in TORCH_PRECOMPUTE_META_FUNC() local110 int64_t poolSizeT, int64_t poolSizeH, int64_t poolSizeW) { in fractional_max_pool3d_out_single_batch_frame()182 int64_t poolSizeT, int64_t poolSizeH, int64_t poolSizeW) { in fractional_max_pool3d_out_frame()
52 int poolSizeH, int poolSizeW) { in fractional_max_pool2d_out_cuda_frame()157 int poolSizeH = pool_size[0]; in TORCH_IMPL_FUNC() local
60 int64_t poolSizeT, int64_t poolSizeH, int64_t poolSizeW) { in fractional_max_pool3d_out_frame()