|
template<typename T , typename TB > |
SimpleTensor< T > | convolution_layer_nchw (const SimpleTensor< T > &src, const SimpleTensor< T > &weights, const SimpleTensor< TB > &bias, SimpleTensor< T > &dst, const PadStrideInfo &info, const Size2D &dilation) |
|
template<typename T , typename TB > |
SimpleTensor< T > | convolution_layer (const SimpleTensor< T > &src, const SimpleTensor< T > &weights, const SimpleTensor< TB > &bias, const TensorShape &output_shape, const PadStrideInfo &info, const Size2D &dilation) |
|
template SimpleTensor< float > | convolution_layer (const SimpleTensor< float > &src, const SimpleTensor< float > &weights, const SimpleTensor< float > &bias, const TensorShape &output_shape, const PadStrideInfo &info, const Size2D &dilation) |
|
template SimpleTensor< half > | convolution_layer (const SimpleTensor< half > &src, const SimpleTensor< half > &weights, const SimpleTensor< half > &bias, const TensorShape &output_shape, const PadStrideInfo &info, const Size2D &dilation) |
|
template SimpleTensor< qint8_t > | convolution_layer (const SimpleTensor< qint8_t > &src, const SimpleTensor< qint8_t > &weights, const SimpleTensor< qint8_t > &bias, const TensorShape &output_shape, const PadStrideInfo &info, const Size2D &dilation) |
|
template SimpleTensor< qint16_t > | convolution_layer (const SimpleTensor< qint16_t > &src, const SimpleTensor< qint16_t > &weights, const SimpleTensor< qint16_t > &bias, const TensorShape &output_shape, const PadStrideInfo &info, const Size2D &dilation) |
|
template SimpleTensor< uint8_t > | convolution_layer (const SimpleTensor< uint8_t > &src, const SimpleTensor< uint8_t > &weights, const SimpleTensor< int32_t > &bias, const TensorShape &output_shape, const PadStrideInfo &info, const Size2D &dilation) |
|