MaxPool.h File Reference

#include "cker/Shape.h"
#include "cker/Utils.h"
#include "cker/eigen/Utils.h"
#include <Eigen/Core>

Go to the source code of this file.

Namespaces
namespace	nnfw
namespace	nnfw::cker
namespace	nnfw::cker::train

Functions
void	nnfw::cker::train::MaxPool2D (const PoolParams &params, const Shape &input_shape, const float *input_data, const Shape &output_shape, float *output_data, int *arg_max_index)
void	nnfw::cker::train::MaxPool2DGrad (const Shape &incoming_shape, const float *incoming_data, const int *arg_max_index, const Shape &grad_shape, float *grad_data)