conv2d.h File Reference

#include <assert.h>
#include <string.h>
#include "tensor.h"

Include dependency graph for conv2d.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Functions
void	NN_nchw_to_nhwc (Tensor *out, const Tensor *in)
void	NN_nhwc_to_nchw (Tensor *out, const Tensor *in)
void	NN_conv2d (Tensor *out, const Tensor *in, const Tensor *weight, const Tensor *bias, const size_t *stride, const size_t *padding, const size_t *dilation, size_t groups)