Searched refs:thnn_conv_use_channels_last (Results 1 – 5 of 5) sorted by relevance
| /external/pytorch/aten/src/ATen/native/ |
| D | NaiveConvolutionTranspose2d.cpp | 212 bool use_channels_last = native::thnn_conv_use_channels_last(input, weight); in TORCH_META_FUNC()
268 bool use_channels_last = thnn_conv_use_channels_last(input, weight); in slow_conv_transpose2d_out_cpu_template()
438 bool use_channels_last = thnn_conv_use_channels_last(input_, weight_); in slow_conv_transpose2d_backward_out_cpu_template()
634 bool use_channels_last = thnn_conv_use_channels_last(input_, weight_); in slow_conv_transpose2d_acc_grad_parameters_cpu()
|
| D | ConvolutionMM2d.cpp | 373 bool use_channels_last = thnn_conv_use_channels_last(input_, weight_); in slow_conv2d_backward_out_cpu_template()
496 bool use_channels_last = thnn_conv_use_channels_last(input, grad_weight); in slow_conv2d_backward_weight_out_cpu_template()
561 bool use_channels_last = thnn_conv_use_channels_last(self, weight_); in slow_conv2d_forward_out_cpu()
|
| D | NaiveDilatedConvolution.cpp | 535 bool use_channels_last = thnn_conv_use_channels_last(input, weight); in slow_conv_dilated2d_cpu()
639 bool use_channels_last = thnn_conv_use_channels_last(input, weight); in slow_conv_dilated2d_backward_cpu()
|
| D | ConvUtils.h | 410 inline bool thnn_conv_use_channels_last(const at::Tensor& input, const at::Tensor& weight) { in thnn_conv_use_channels_last() function
|
| D | Convolution.cpp | 1445 if (thnn_conv_use_channels_last(input, weight)) { in determine_backend_memory_format()
|