ONE - On-device Neural Engine
Loading...
Searching...
No Matches
training_ops.h File Reference
#include "unsupported/Eigen/CXX11/Tensor"
#include "cker/operation/Helper/Tensor.h"

Go to the source code of this file.

Data Structures

struct  nnfw::cker::training_ops::functor::ApplyAdam< Device, T >
 
struct  nnfw::cker::training_ops::functor::ApplyGradientDescent< Device, T >
 
struct  nnfw::cker::training_ops::functor::ApplyAdamNonCuda< Device, T >
 
struct  nnfw::cker::training_ops::functor::ApplyAdam< CPUDevice, T >
 
struct  nnfw::cker::training_ops::functor::ApplyGradientDescent< CPUDevice, T >
 

Namespaces

namespace  nnfw
 
namespace  nnfw::cker
 
namespace  nnfw::cker::training_ops
 
namespace  nnfw::cker::training_ops::functor
 

Macros

#define EIGEN_USE_THREADS
 

Typedefs

using nnfw::cker::training_ops::CPUDevice = Eigen::ThreadPoolDevice
 
using nnfw::cker::training_ops::Index = Eigen::Index
 

Macro Definition Documentation

◆ EIGEN_USE_THREADS

#define EIGEN_USE_THREADS

Definition at line 22 of file training_ops.h.