ONE - On-device Neural Engine
Loading...
Searching...
No Matches
onert::exec::train::IGradientApplier Class Referenceabstract

#include <IGradientApplier.h>

Public Member Functions

virtual ~IGradientApplier ()=default
 
virtual void applyGradient (uint32_t training_step)=0
 Apply gradients to a trainable tensor.
 

Detailed Description

Definition at line 29 of file IGradientApplier.h.

Constructor & Destructor Documentation

◆ ~IGradientApplier()

virtual onert::exec::train::IGradientApplier::~IGradientApplier ( )
virtualdefault

Member Function Documentation

◆ applyGradient()

virtual void onert::exec::train::IGradientApplier::applyGradient ( uint32_t  training_step)
pure virtual

Apply gradients to a trainable tensor.

Parameters
training_stepThe number of iterations of the training process.

Implemented in onert::backend::train::ops::GradientApplier.


The documentation for this class was generated from the following file: