24 #ifndef ARM_COMPUTE_TEST_CONVOLUTION_LAYER_H 25 #define ARM_COMPUTE_TEST_CONVOLUTION_LAYER_H 38 template <
typename T,
typename TW,
typename TB>
39 SimpleTensor<T>
convolution_layer(
const SimpleTensor<T> &
src,
const SimpleTensor<TW> &weights,
const SimpleTensor<TB> &bias,
const TensorShape &
output_shape,
const PadStrideInfo &
info,
40 const Size2D &dilation = Size2D(1U, 1U),
unsigned int num_groups = 1, QuantizationInfo out_quant_info = QuantizationInfo());
SimpleTensor< float > src
Copyright (c) 2017-2021 Arm Limited.
const unsigned int num_groups
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
SimpleTensor< T > convolution_layer(const SimpleTensor< T > &src, const SimpleTensor< TW > &weights, const SimpleTensor< TB > &bias, const TensorShape &output_shape, const PadStrideInfo &info, const Size2D &dilation, unsigned int num_groups, QuantizationInfo out_quant_info)