Searched defs:dilation_expanded (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | Conv.cpp | 242 const auto dilation_expanded = expand_param_if_needed(dilation, "dilation", dim); in _mkldnn_convolution() local
367 const auto dilation_expanded = expand_param_if_needed(dilation, "dilation", dim); in mkldnn_convolution_pointwise_binary() local
537 const auto dilation_expanded = expand_param_if_needed(dilation, "dilation", dim); in mkldnn_convolution_pointwise_binary_() local
656 const auto dilation_expanded = expand_param_if_needed(dilation, "dilation", dim); in _mkldnn_convolution_transpose() local
762 const auto dilation_expanded = expand_param_if_needed(dilation, "dilation", dim); in mkldnn_convolution_transpose_pointwise_meta() local
878 const auto dilation_expanded = expand_param_if_needed(dilation, "dilation", dim); in mkldnn_convolution_backward() local
1029 const auto dilation_expanded = expand_param_if_needed(dilation, "dilation", dim); in mkldnn_convolution_transpose_backward() local
|
| H A D | MKLDNNConversions.cpp | 151 const auto dilation_expanded = expand_param_if_needed(dilation, "dilation", 2); in mkldnn_reorder_conv2d_weight() local
208 const auto dilation_expanded = expand_param_if_needed(dilation, "dilation", 3); in mkldnn_reorder_conv3d_weight() local
340 const auto dilation_expanded = in mkldnn_reorder_conv_transpose_weight() local
|
| H A D | ConvPrepack.cpp | 57 const auto dilation_expanded = in create() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/xnnpack/ |
| H A D | Convolution.cpp | 185 const auto dilation_expanded = expand_param_if_needed(dilation, "dilation", 2); in create() local
|