Searched defs:dimh (Results 1 – 12 of 12) sorted by relevance
95 int64_t dimh = 1; in TORCH_META_FUNC() local137 int64_t dimh = 2; in TORCH_META_FUNC() local192 int dimh = 1; in replication_pad2d_backward_out_cpu_template() local233 int dimh = 2; in replication_pad3d_backward_out_cpu_template() local
123 int dimh = 2; in max_unpooling3d_shape_check() local
93 int dimh = 1; in slow_conv_transpose2d_shape_check() local
116 int dimh = 2; in slow_conv_transpose3d_shape_check() local
153 int64_t dimh = 1; in max_unpooling2d_forward_out_cuda() local254 int dimh = 2; in max_unpooling3d_shape_check() local432 int dimh = 1; in max_unpooling2d_backward_out_cuda() local
245 int dimh = 1; in replication_pad2d_backward_out_cuda_template() local323 int dimh = 2; in shapeAndGradOutputCheck3d() local614 int dimh = 2; in TORCH_IMPL_FUNC() local
47 constexpr int dimh = 2; in slow_conv2d_shape_check() local138 constexpr int dimh = 2; in slow_conv2d_forward() local
215 int dimh = 1; in TORCH_IMPL_FUNC() local
162 int64_t dimh = 2; in fractional_max_pool3d_backward_out_cuda_template() local
96 int dimh = 1; in slow_conv_transpose2d_shape_check() local
115 int dimh = 2; in slow_conv_transpose3d_shape_check() local
214 int dimh = 1; in q_maxpool_2d() local386 int dimh = 3; in q_maxpool_3d() local