Searched defs:stride_expanded (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | Conv.cpp | 241 const auto stride_expanded = expand_param_if_needed(stride, "stride", dim); in _mkldnn_convolution() local
366 const auto stride_expanded = expand_param_if_needed(stride, "stride", dim); in mkldnn_convolution_pointwise_binary() local
536 const auto stride_expanded = expand_param_if_needed(stride, "stride", dim); in mkldnn_convolution_pointwise_binary_() local
655 const auto stride_expanded = expand_param_if_needed(stride, "stride", dim); in _mkldnn_convolution_transpose() local
761 const auto stride_expanded = expand_param_if_needed(stride, "stride", dim); in mkldnn_convolution_transpose_pointwise_meta() local
877 const auto stride_expanded = expand_param_if_needed(stride, "stride", dim); in mkldnn_convolution_backward() local
1028 const auto stride_expanded = expand_param_if_needed(stride, "stride", dim); in mkldnn_convolution_transpose_backward() local
|
| H A D | MKLDNNConversions.cpp | 150 const auto stride_expanded = expand_param_if_needed(stride, "stride", 2); in mkldnn_reorder_conv2d_weight() local
207 const auto stride_expanded = expand_param_if_needed(stride, "stride", 3); in mkldnn_reorder_conv3d_weight() local
339 const auto stride_expanded = expand_param_if_needed(stride, "stride", pdim); in mkldnn_reorder_conv_transpose_weight() local
|
| H A D | ConvPrepack.cpp | 56 const auto stride_expanded = expand_param_if_needed(stride, "stride", dim); in create() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/xnnpack/ |
| H A D | Convolution.cpp | 184 const auto stride_expanded = expand_param_if_needed(stride, "stride", 2); in create() local
|