Searched refs:padding_expanded (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/aten/src/ATen/native/mkldnn/ |
| D | Conv.cpp | 240 const auto padding_expanded = expand_param_if_needed(padding, "padding", dim); in _mkldnn_convolution() local
244 …check_shape_forward(input_t, weight_t, bias, padding_expanded, stride_expanded, dilation_expanded,… in _mkldnn_convolution()
249 …auto output_sizes = conv_output_size(input_t.sizes(), weight_t.sizes(), padding_expanded, stride_e… in _mkldnn_convolution()
264 padding_expanded, in _mkldnn_convolution()
365 const auto padding_expanded = expand_param_if_needed(padding, "padding", dim); in mkldnn_convolution_pointwise_binary() local
369 input_t, weight_t, bias, padding_expanded, stride_expanded, dilation_expanded, groups); in mkldnn_convolution_pointwise_binary()
372 input_t.sizes(), weight_t.sizes(), padding_expanded, stride_expanded, dilation_expanded); in mkldnn_convolution_pointwise_binary()
439 padding_expanded, in mkldnn_convolution_pointwise_binary()
440 padding_expanded, in mkldnn_convolution_pointwise_binary()
453 padding_expanded, in mkldnn_convolution_pointwise_binary()
[all …]
|
| D | ConvPrepack.cpp | 55 const auto padding_expanded = expand_param_if_needed(padding, "padding", dim); in create() local
72 {padding_expanded.begin(), padding_expanded.end()}, in create()
73 {padding_expanded.begin(), padding_expanded.end()}, in create()
90 {padding_expanded.begin(), padding_expanded.end()}, in create()
|
| D | MKLDNNConversions.cpp | 149 const auto padding_expanded = expand_param_if_needed(padding, "padding", 2); in mkldnn_reorder_conv2d_weight() local
180 padding_expanded, in mkldnn_reorder_conv2d_weight()
181 padding_expanded, in mkldnn_reorder_conv2d_weight()
206 const auto padding_expanded = expand_param_if_needed(padding, "padding", 3); in mkldnn_reorder_conv3d_weight() local
227 padding_expanded, in mkldnn_reorder_conv3d_weight()
228 padding_expanded, in mkldnn_reorder_conv3d_weight()
337 const auto padding_expanded = in mkldnn_reorder_conv_transpose_weight() local
363 padding_expanded, in mkldnn_reorder_conv_transpose_weight()
364 padding_r(padding_expanded, output_padding_expanded), in mkldnn_reorder_conv_transpose_weight()
|
| /external/pytorch/aten/src/ATen/native/xnnpack/ |
| D | Convolution.cpp | 182 const auto padding_expanded = expand_param_if_needed(padding, "padding", 2); in create() local
192 padding_expanded, in create()
216 padding_expanded[Layout::Parameter::height], // output_padding_top in create()
217 padding_expanded[Layout::Parameter::width], // output_padding_right in create()
218 padding_expanded[Layout::Parameter::height], // output_padding_bottom in create()
219 padding_expanded[Layout::Parameter::width], // output_padding_left in create()
246 padding_expanded[Layout::Parameter::height], // input_padding_top in create()
247 padding_expanded[Layout::Parameter::width], // input_padding_right in create()
248 padding_expanded[Layout::Parameter::height], // input_padding_bottom in create()
249 padding_expanded[Layout::Parameter::width], // input_padding_left in create()
[all …]
|