ONE - On-device Neural Engine
All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Modules Pages
ReduceMean.h File Reference

Go to the source code of this file.

Namespaces

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

Functions

template<typename T >
void nnfw::cker::train::MeanGrad (const Shape &incoming_shape, const T *incoming_data, const Shape &grad_shape, T *grad_data)