Searched refs:mkldnn_conv_use_channels_last (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/aten/src/ATen/native/mkldnn/ |
| D | Conv.cpp | 286 bool use_channels_last = mkldnn_conv_use_channels_last(input_t, weight_t); in mkldnn_convolution()
312 weight_t.is_mkldnn() || mkldnn_conv_use_channels_last(input_t, weight_t); in mkldnn_convolution_pointwise()
381 weight_t.is_mkldnn() || mkldnn_conv_use_channels_last(input_t, weight_t); in mkldnn_convolution_pointwise_binary()
548 mkldnn_conv_use_channels_last(input_t, weight_t)) && in mkldnn_convolution_pointwise_binary_()
728 weight_t.is_mkldnn() || mkldnn_conv_use_channels_last(input_t, weight_t); in mkldnn_convolution_transpose_pointwise()
869 bool is_channels_last = mkldnn_conv_use_channels_last(input_t, weight_t); in mkldnn_convolution_backward()
905 bool use_channels_last = mkldnn_conv_use_channels_last(input, weight); in mkldnn_convolution_transpose()
1021 bool is_channels_last = mkldnn_conv_use_channels_last(input_t, weight_t); in mkldnn_convolution_transpose_backward()
|
| /external/pytorch/aten/src/ATen/native/ |
| D | ConvUtils.h | 383 inline bool mkldnn_conv_use_channels_last(const at::Tensor& input, const at::Tensor& weight) { in mkldnn_conv_use_channels_last() function
|
| D | Convolution.cpp | 1438 if (mkldnn_conv_use_channels_last(input, weight)) { in determine_backend_memory_format()
|