conv2d.h

Go to the documentation of this file.

#ifndef __NN_conv2d_H
#define __NN_conv2d_H
 
#include <assert.h>
#include <string.h>
 
#include "tensor.h"
 
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
  );
 
 
#endif // __NN_conv2d_H