Searched refs:input_padding (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/aten/src/ATen/native/ |
| D | NNPACK.cpp | 206 const struct nnp_padding input_padding = { in _nnpack_spatial_convolution() local
242 input_padding, in _nnpack_spatial_convolution()
270 input_padding, in _nnpack_spatial_convolution()
|
| /external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/ |
| D | depthwise_depthfirst_multiplier.hpp | 258 T *input_padding; member
275 ws->input_padding = reinterpret_cast<T*>(buffer_bytes); in initialise()
282 memset(ws->input_padding, in initialise()
438 ws->input_padding, in execute()
461 ws->input_padding, in execute()
|
| /external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| D | qconv.cpp | 100 int64_t input_padding, in compute_deconv_shape() argument
103 int64_t out = (input - 1) * stride - 2 * input_padding in compute_deconv_shape()
114 const torch::List<int64_t>& input_padding, in MakeDeConvOutputShape() argument
125 input_padding[idx], in MakeDeConvOutputShape()
132 ", input padding: ", input_padding[idx], in MakeDeConvOutputShape()
140 ", input padding: ", input_padding[idx], in MakeDeConvOutputShape()
|