Searched defs:poolSizeW (Results 1 – 4 of 4) sorted by relevance
39 int64_t poolSizeW = pool_size[1]; 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()299 int64_t poolSizeW = pool_size[1]; in TORCH_IMPL_FUNC() local
38 int64_t poolSizeW = pool_size[2]; 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()158 int poolSizeW = pool_size[1]; in TORCH_IMPL_FUNC() local
60 int64_t poolSizeT, int64_t poolSizeH, int64_t poolSizeW) { in fractional_max_pool3d_out_frame()