Searched defs:n_input_plane (Results 1 – 12 of 12) sorted by relevance
46 int64_t n_input_plane, in unfolded2d_acc()131 int64_t n_input_plane, in unfolded2d_acc_channels_last()188 int64_t n_input_plane, in unfolded2d_acc_kernel()239 int64_t n_input_plane, in unfolded2d_copy()346 int64_t n_input_plane, in unfolded2d_copy_channels_last()403 int64_t n_input_plane, in unfolded2d_copy_kernel()
41 const int64_t n_input_plane = input.size(1); in compute_columns2d() local184 int64_t n_input_plane = weight.size(1); in slow_conv2d_shape_check() local234 int64_t n_input_plane, in slow_conv2d_update_output_frame()394 const int64_t n_input_plane = input.size(1); in slow_conv2d_backward_out_cpu_template() local581 const int64_t n_input_plane = input.size(1); in slow_conv2d_forward_out_cpu() local
129 int64_t n_input_plane = weight.size(0); in slow_conv_transpose2d_shape_check() local265 int n_input_plane = weight.size(0); in slow_conv_transpose2d_out_cpu_template() local435 int64_t n_input_plane = weight_.size(0); in slow_conv_transpose2d_backward_out_cpu_template() local654 int n_input_plane = weight_.size(0); in slow_conv_transpose2d_acc_grad_parameters_cpu() local
50 const int64_t n_input_plane = input.size(dim_planes); in compute_columns3d() local225 int64_t n_input_plane = weight.size(1); in slow_conv3d_shape_check() local278 int64_t n_input_plane, in slow_conv3d_update_output_frame()415 const int64_t n_input_plane = input.size(dim_planes); in slow_conv3d_backward_out_cpu_template() local603 const int64_t n_input_plane = input.size(dim_planes); in slow_conv3d_forward_out_cpu() local
127 const int64_t n_input_plane = weight.size(0); in slow_conv_transpose3d_shape_check() local254 const int n_input_plane = (int)weight.size(0); in slow_conv_transpose3d_out_cpu_template() local470 const int64_t n_input_plane = weight.size(0); in slow_conv_transpose3d_backward_out_cpu_template() local
56 int64_t n_input_plane = input.size(batch_dim + 1); in col2im_shape_check() local
80 int64_t n_input_plane = input.size(1); in im2col_out_cpu_template() local
142 int64_t n_input_plane = input.size(1); in col2im_out_cpu_template() local
131 int64_t n_input_plane = weight.size(0); in slow_conv_transpose2d_shape_check() local165 int n_input_plane = weight.size(0); in slow_conv_transpose2d_out_cuda_template() local350 int n_input_plane = weight_.size(0); in slow_conv_transpose2d_backward_out_cuda_template() local
126 const int64_t n_input_plane = weight.size(0); in slow_conv_transpose3d_shape_check() local224 int n_input_plane = weight_.size(0); in slow_conv_transpose3d_out_cuda_template() local429 int n_input_plane = weight_.size(0); in slow_conv_transpose3d_backward_out_cuda_template() local
88 int64_t n_input_plane = input.size(1); in im2col_out_cuda_template() local
98 int64_t n_input_plane = input.size(1); in col2im_out_cuda_template() local