Searched defs:nOutputPlane (Results 1 – 4 of 4) sorted by relevance
102 int64_t nOutputPlane = b_sizes.size() == 0 ? 1 : b_sizes[0]; in slow_conv2d_shape_check() local146 int64_t nOutputPlane = weight.sizes()[0]; in slow_conv2d_forward() local232 int nOutputPlane = weight_sizes[0]; in slow_conv2d_backward() local310 int64_t nOutputPlane = grad_output.sizes()[1]; in slow_conv2d_grad_weight() local
203 int64_t nOutputPlane = weight.size(0); in slow_conv_dilated_all_cuda_template() local
130 const int64_t nOutputPlane = nInputPlane; in pool2d_shape_check() local185 const int64_t nOutputPlane = nInputPlane; in max_pool2d_backward_shape_check() local215 const int64_t nOutputPlane = nInputPlane; in avg_pool2d_backward_shape_check() local
195 int64_t nOutputPlane = weight.size(0); in slow_conv_dilated_all_cpu_template() local