Searched defs:n_output_plane (Results 1 – 10 of 10) sorted by relevance
135 int64_t n_output_plane = weight.size(1); in slow_conv_transpose2d_shape_check() local138 int64_t n_output_plane = bias.size(0); in slow_conv_transpose2d_shape_check() local210 int n_output_plane = weight.size(1); in TORCH_META_FUNC() local266 int n_output_plane = weight.size(1); in slow_conv_transpose2d_out_cpu_template() local436 int64_t n_output_plane = weight_.size(1); in slow_conv_transpose2d_backward_out_cpu_template() local655 int n_output_plane = weight_.size(1); in slow_conv_transpose2d_acc_grad_parameters_cpu() local
164 const int64_t n_output_plane = weight.size(1); in slow_conv_transpose3d_shape_check() local167 const int64_t n_output_plane = bias.size(0); in slow_conv_transpose3d_shape_check() local255 const int n_output_plane = (int)weight.size(1); in slow_conv_transpose3d_out_cpu_template() local471 const int64_t n_output_plane = weight.size(1); in slow_conv_transpose3d_backward_out_cpu_template() local672 int64_t n_output_plane; in slow_conv_transpose3d_acc_grad_parameters_cpu() local
195 int64_t n_output_plane = weight.size(0); in slow_conv2d_shape_check() local199 const int64_t n_output_plane = bias.dim() == 0 ? 1 : bias.size(0); in slow_conv2d_shape_check() local237 int64_t n_output_plane, in slow_conv2d_update_output_frame()584 const int64_t n_output_plane = weight_2d.size(0); in slow_conv2d_forward_out_cpu() local
237 int64_t n_output_plane = weight.size(0); in slow_conv3d_shape_check() local241 const int64_t n_output_plane = bias.dim() == 0 ? 1 : bias.size(0); in slow_conv3d_shape_check() local283 int64_t n_output_plane, in slow_conv3d_update_output_frame()607 const int64_t n_output_plane = weight_2d.size(0); in slow_conv3d_forward_out_cpu() local
92 int64_t n_output_plane = n_input_plane * kernel_width * kernel_height; in im2col_out_cpu_template() local
143 int64_t n_output_plane = n_input_plane / (kernel_width * kernel_height); in col2im_out_cpu_template() local
137 int64_t n_output_plane = weight.size(1); in slow_conv_transpose2d_shape_check() local140 int64_t n_output_plane = bias.size(0); in slow_conv_transpose2d_shape_check() local166 int n_output_plane = weight.size(1); in slow_conv_transpose2d_out_cuda_template() local351 int n_output_plane = weight_.size(1); in slow_conv_transpose2d_backward_out_cuda_template() local560 int64_t n_output_plane; in slow_conv_transpose2d_acc_grad_parameters_cuda_template() local
161 const int64_t n_output_plane = weight.size(1); in slow_conv_transpose3d_shape_check() local164 const int64_t n_output_plane = bias.size(0); in slow_conv_transpose3d_shape_check() local225 int n_output_plane = weight_.size(1); in slow_conv_transpose3d_out_cuda_template() local430 int n_output_plane = weight_.size(1); in slow_conv_transpose3d_backward_out_cuda_template() local690 int n_output_plane; in slow_conv_transpose3d_acc_grad_parameters_cuda() local
100 int64_t n_output_plane = n_input_plane * kernel_width * kernel_height; in im2col_out_cuda_template() local
99 int64_t n_output_plane = n_input_plane / (kernel_width * kernel_height); in col2im_out_cuda_template() local