Searched refs:mkldnn_convolution_memory_format (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/aten/src/ATen/native/mkldnn/ |
| D | Conv.cpp | 157 static inline at::MemoryFormat mkldnn_convolution_memory_format(int64_t dims, bool is_channels_last… in mkldnn_convolution_memory_format() function
177 auto memory_format = mkldnn_convolution_memory_format(input_t.ndimension(), is_channels_last); in _mkldnn_convolution_out()
247 mkldnn_convolution_memory_format(input_t.ndimension(), use_channels_last); in _mkldnn_convolution()
402 mkldnn_convolution_memory_format(input_t.ndimension(), true); in mkldnn_convolution_pointwise_binary()
648 mkldnn_convolution_memory_format(input_t.ndimension(), use_channels_last); in _mkldnn_convolution_transpose()
787 … auto memory_format = mkldnn_convolution_memory_format(grad_output.ndimension(), is_channels_last); in mkldnn_convolution_backward_input()
858 … auto memory_format = mkldnn_convolution_memory_format(grad_output.ndimension(), is_channels_last); in mkldnn_convolution_backward_weights()
870 auto memory_format = mkldnn_convolution_memory_format(input_t.ndimension(), is_channels_last); in mkldnn_convolution_backward()
937 … auto memory_format = mkldnn_convolution_memory_format(grad_output.ndimension(), is_channels_last); in mkldnn_convolution_transpose_backward_input()
1009 … auto memory_format = mkldnn_convolution_memory_format(grad_output.ndimension(), is_channels_last); in mkldnn_convolution_transpose_backward_weights()
[all …]
|