Searched defs:outputPlaneSize (Results 1 – 3 of 3) sorted by relevance
283 int64_t outputPlaneSize = devGradOutput.size(2) * devGradOutput.size(3); in replication_pad2d_backward_out_cuda_template() local397 …const int64_t outputPlaneSize = devGradOutput.size(2) * devGradOutput.size(3) * devGradOutput.size… in replication_pad3d_backward_out_cuda_template() local452 int64_t outputPlaneSize = devOutput.size(2); in TORCH_IMPL_FUNC() local514 int64_t outputPlaneSize = devGradOutput.size(2); in TORCH_IMPL_FUNC() local556 int64_t outputPlaneSize = devOutput.size(2) * devOutput.size(3); in TORCH_IMPL_FUNC() local650 const int64_t outputPlaneSize = devOutput.size(2) * devOutput.size(3) * devOutput.size(4); in TORCH_IMPL_FUNC() local
176 int outputPlaneSize = output_.size(2) * in TORCH_IMPL_FUNC() local250 int outputPlaneSize = gradOutput_.size(2) * in TORCH_IMPL_FUNC() local
221 int64_t outputPlaneSize = gradOutput_.size(2) * in fractional_max_pool3d_backward_out_cuda_template() local282 int64_t outputPlaneSize = output_.size(2) * in TORCH_IMPL_FUNC() local