#include <BackPropInitializer.h>

Collaboration diagram for onert::backend::train::ops::BackPropInitializer:

Public Member Functions
	BackPropInitializer (const std::vector< BackPropTensor * > back_props)

void	forward (bool training) override

void	backward () override

Public Member Functions inherited from onert::exec::train::ITrainableFunction
virtual	~ITrainableFunction ()=default

virtual std::optional< backend::train::LayerScopeTensors >	registerLayerScopeTensors ()

Detailed Description

Definition at line 35 of file BackPropInitializer.h.

Constructor & Destructor Documentation

◆ BackPropInitializer()

onert::backend::train::ops::BackPropInitializer::BackPropInitializer ( const std::vector< BackPropTensor * > back_props )

Definition at line 28 of file BackPropInitializer.cc.

  : _back_props{back_props}
{
  assert(std::all_of(back_props.cbegin(), back_props.cend(),
                     [](const BackPropTensor *back_prop) { return back_prop != nullptr; }));
}

Member Function Documentation

◆ backward()

void onert::backend::train::ops::BackPropInitializer::backward ( )

overridevirtual

Implements onert::exec::train::ITrainableFunction.

Definition at line 40 of file BackPropInitializer.cc.

{
  for (auto &&back_prop_tensor : _back_props)
  {
    assert(back_prop_tensor->buffer() != nullptr);
    memset(back_prop_tensor->buffer(), 0, back_prop_tensor->total_size());
  }
}

◆ forward()

void onert::backend::train::ops::BackPropInitializer::forward ( bool training )