Searched defs:dimw (Results 1 – 12 of 12) sorted by relevance
30 int64_t dimw = 1; in TORCH_META_FUNC() local66 int64_t dimw = 1; in TORCH_META_FUNC() local94 int64_t dimw = 2; in TORCH_META_FUNC() local136 int64_t dimw = 3; in TORCH_META_FUNC() local191 int dimw = 2; in replication_pad2d_backward_out_cpu_template() local232 int dimw = 3; in replication_pad3d_backward_out_cpu_template() local
122 int dimw = 3; in max_unpooling3d_shape_check() local
94 int dimw = 2; in slow_conv_transpose2d_shape_check() local
117 int dimw = 3; in slow_conv_transpose3d_shape_check() local
246 int dimw = 2; in replication_pad2d_backward_out_cuda_template() local324 int dimw = 3; in shapeAndGradOutputCheck3d() local428 constexpr int64_t dimw = -1; in TORCH_IMPL_FUNC() local489 int64_t dimw = 1; in TORCH_IMPL_FUNC() local615 int dimw = 3; in TORCH_IMPL_FUNC() local
152 int64_t dimw = 2; in max_unpooling2d_forward_out_cuda() local253 int dimw = 3; in max_unpooling3d_shape_check() local431 int dimw = 2; in max_unpooling2d_backward_out_cuda() local
48 constexpr int dimw = 3; in slow_conv2d_shape_check() local139 constexpr int dimw = 3; in slow_conv2d_forward() local
216 int dimw = 2; in TORCH_IMPL_FUNC() local
163 int64_t dimw = 3; in fractional_max_pool3d_backward_out_cuda_template() local
97 int dimw = 2; in slow_conv_transpose2d_shape_check() local
116 int dimw = 3; in slow_conv_transpose3d_shape_check() local
215 int dimw = 2; in q_maxpool_2d() local387 int dimw = 4; in q_maxpool_3d() local