|
Baremetal-NN
Baremetal-NN API documentation
|
| 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
| out | The output tensor. |
| in | The input tensor. |
| weight | The weight tensor. |