ONE - On-device Neural Engine
Loading...
Searching...
No Matches
ReLU.h File Reference
#include "cker/Shape.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::ReLUGrad (const Shape &output_shape, const float *output_data, const Shape &incoming_shape, const float *incoming_data, const Shape &grad_shape, float *grad_data)