Baremetal-NN
Baremetal-NN API documentation
Loading...
Searching...
No Matches

◆ nn_conv2d_f32()

void nn_conv2d_f32 ( Tensor4D_F32 out,
const Tensor4D_F32 in,
const Tensor4D_F32 weight,
const Tensor1D_F32 bias,
const size_t *  stride,
const size_t *  padding,
const size_t *  dilation,
size_t  groups 
)

Performs a 2D convolution on a 4D tensor.

nn_conv2d_f32

Parameters
outThe output tensor.
inThe input tensor.
weightThe weight tensor.