| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/xpu/detail/ |
| H A D | Conv.cpp | 53 const bool is_channels_last = false) { in conv_src_fmt()
72 const bool is_channels_last = false) { in conv_weight_fmt()
129 bool is_channels_last) { in conv_get_md()
174 bool is_channels_last = use_channels_last_for_conv(src, weight, false); in convolution() local
268 bool is_channels_last = use_channels_last_for_conv(src, diff_dst, /*is_transposed=*/false); in convolution_backward_weights() local
369 bool is_channels_last = use_channels_last_for_conv(diff_dst, weight, /*is_transposed=*/false); in convolution_backward_data() local
|
| H A D | Utils.cpp | 18 bool is_channels_last, in get_dnnl_default_format()
352 static inline bool is_channels_last(at::MemoryFormat fmt){ in is_channels_last() function
|
| H A D | Deconv.cpp | 52 const bool is_channels_last = false) { in deconv_src_fmt()
76 const bool is_channels_last = false) { in deconv_weight_fmt()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | Conv.cpp | 157 static inline at::MemoryFormat mkldnn_convolution_memory_format(int64_t dims, bool is_channels_last… in mkldnn_convolution_memory_format()
175 bool is_channels_last, in _mkldnn_convolution_out()
779 bool is_channels_last) { in mkldnn_convolution_backward_input()
821 bool is_channels_last) { in mkldnn_convolution_backward_weights()
869 bool is_channels_last = mkldnn_conv_use_channels_last(input_t, weight_t); in mkldnn_convolution_backward() local
929 bool is_channels_last) { in mkldnn_convolution_transpose_backward_input()
972 bool is_channels_last) { in mkldnn_convolution_transpose_backward_weights()
1021 bool is_channels_last = mkldnn_conv_use_channels_last(input_t, weight_t); in mkldnn_convolution_transpose_backward() local
|
| H A D | ConvPrepack.cpp | 65 bool is_channels_last = in create() local
201 bool is_channels_last = in run() local
240 bool is_channels_last = in run() local
|
| H A D | MKLDNNConversions.cpp | 154 bool is_channels_last = false; in mkldnn_reorder_conv2d_weight() local
211 bool is_channels_last = false; in mkldnn_reorder_conv3d_weight() local
346 bool is_channels_last = false; in mkldnn_reorder_conv_transpose_weight() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | NaiveDilatedConvolution.cpp | 38 bool is_channels_last = false) { in hvol2col()
96 bool is_channels_last = false) { in col2hvol()
185 bool is_channels_last = false) { in slow_conv_dilated_all_cpu_template()
|
| H A D | ConvolutionMM2d.cpp | 33 bool is_channels_last) { in compute_columns2d()
240 bool is_channels_last) { in slow_conv2d_update_output_frame()
297 bool is_channels_last) { in slow_conv2d_backward_update_grad_input_frame()
439 bool is_channels_last) { in slow_conv2d_backward_weight_frame()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | Unfold2d.cpp | 193 bool is_channels_last) { in unfolded2d_acc_kernel()
408 bool is_channels_last) { in unfolded2d_copy_kernel()
|
| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | SymbolicShapeMeta.h | 104 const SymBool& is_channels_last() const { in is_channels_last() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | qnormalization.cpp | 64 bool is_channels_last = qx.is_contiguous(c10::MemoryFormat::ChannelsLast); in quantized_group_norm_impl() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | TensorIterator.cpp | 1440 bool is_channels_last = true; in compute_fast_setup_type() local
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 2226 def is_channels_last(ten): function
6133 def is_channels_last(ten): function
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_nn.py | 6477 def helper(size, dtype, mode, device, is_channels_last): argument
|