max_pool2d.h

Go to the documentation of this file.

#ifndef __NN_MAX_POOL2D_H
#define __NN_MAX_POOL2D_H
 
#include <assert.h>
 
#include "tensor.h"
#include "max.h"
 
 
void NN_max_pool2d(Tensor *out, const Tensor *in, const size_t *kernel_size);
 
 
#endif // __NN_MAX_POOL2D_H