Searched defs:padding_expanded (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | Conv.cpp | 240 const auto padding_expanded = expand_param_if_needed(padding, "padding", dim); in _mkldnn_convolution() local
365 const auto padding_expanded = expand_param_if_needed(padding, "padding", dim); in mkldnn_convolution_pointwise_binary() local
535 const auto padding_expanded = expand_param_if_needed(padding, "padding", dim); in mkldnn_convolution_pointwise_binary_() local
654 const auto padding_expanded = expand_param_if_needed(padding, "padding", dim); in _mkldnn_convolution_transpose() local
760 const auto padding_expanded = expand_param_if_needed(padding, "padding", dim); in mkldnn_convolution_transpose_pointwise_meta() local
876 const auto padding_expanded = expand_param_if_needed(padding, "padding", dim); in mkldnn_convolution_backward() local
1027 const auto padding_expanded = expand_param_if_needed(padding, "padding", dim); in mkldnn_convolution_transpose_backward() local
|
| H A D | MKLDNNConversions.cpp | 149 const auto padding_expanded = expand_param_if_needed(padding, "padding", 2); in mkldnn_reorder_conv2d_weight() local
206 const auto padding_expanded = expand_param_if_needed(padding, "padding", 3); in mkldnn_reorder_conv3d_weight() local
337 const auto padding_expanded = in mkldnn_reorder_conv_transpose_weight() local
|
| H A D | ConvPrepack.cpp | 55 const auto padding_expanded = expand_param_if_needed(padding, "padding", dim); in create() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/xnnpack/ |
| H A D | Convolution.cpp | 182 const auto padding_expanded = expand_param_if_needed(padding, "padding", 2); in create() local
|