Searched defs:use_channels_last (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | Conv.cpp | 221 bool use_channels_last, in _mkldnn_convolution()
286 bool use_channels_last = mkldnn_conv_use_channels_last(input_t, weight_t); in mkldnn_convolution() local
311 bool use_channels_last = in mkldnn_convolution_pointwise() local
380 bool use_channels_last = in mkldnn_convolution_pointwise_binary() local
627 bool use_channels_last, in _mkldnn_convolution_transpose()
727 bool use_channels_last = in mkldnn_convolution_transpose_pointwise() local
905 bool use_channels_last = mkldnn_conv_use_channels_last(input, weight); in mkldnn_convolution_transpose() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | NaiveConvolutionTranspose2d.cpp | 212 bool use_channels_last = native::thnn_conv_use_channels_last(input, weight); in TORCH_META_FUNC() local
268 bool use_channels_last = thnn_conv_use_channels_last(input, weight); in slow_conv_transpose2d_out_cpu_template() local
438 bool use_channels_last = thnn_conv_use_channels_last(input_, weight_); in slow_conv_transpose2d_backward_out_cpu_template() local
634 bool use_channels_last = thnn_conv_use_channels_last(input_, weight_); in slow_conv_transpose2d_acc_grad_parameters_cpu() local
|
| H A D | ConvolutionMM2d.cpp | 373 bool use_channels_last = thnn_conv_use_channels_last(input_, weight_); in slow_conv2d_backward_out_cpu_template() local
496 bool use_channels_last = thnn_conv_use_channels_last(input, grad_weight); in slow_conv2d_backward_weight_out_cpu_template() local
561 bool use_channels_last = thnn_conv_use_channels_last(self, weight_); in slow_conv2d_forward_out_cpu() local
|
| H A D | NaiveDilatedConvolution.cpp | 535 bool use_channels_last = thnn_conv_use_channels_last(input, weight); in slow_conv_dilated2d_cpu() local
639 bool use_channels_last = thnn_conv_use_channels_last(input, weight); in slow_conv_dilated2d_backward_cpu() local
|