onert::backend::train::ops::LossLayer Class Reference

#include <LossLayer.h>

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

Public Member Functions
	LossLayer ()
void	configure (const IPortableTensor *y_pred, const IPortableTensor *y_true, IPortableTensor *output, IPortableTensor *back_prop_y_pred, ir::train::LossReductionType reduction_type)
Public Member Functions inherited from onert::exec::train::ITrainableFunction
virtual	~ITrainableFunction ()=default
virtual void	forward (bool training)=0
virtual void	backward ()=0
virtual std::optional< backend::train::LayerScopeTensors >	registerLayerScopeTensors ()

Protected Attributes
const IPortableTensor *	_y_pred
const IPortableTensor *	_y_true
IPortableTensor *	_output
IPortableTensor *	_back_prop_y_pred
ir::train::LossReductionType	_reduction_type

Detailed Description

Definition at line 40 of file LossLayer.h.

Constructor & Destructor Documentation

LossLayer()

onert::backend::train::ops::LossLayer::LossLayer

(

)

Definition at line 28 of file LossLayer.cc.

  : _y_pred(nullptr), _y_true(nullptr), _output(nullptr), _back_prop_y_pred(nullptr),
    _reduction_type(ir::train::LossReductionType::Undefined)
{
  // DO NOTHING
}

Member Function Documentation

configure()

void onert::backend::train::ops::LossLayer::configure	(	const IPortableTensor *	y_pred,
		const IPortableTensor *	y_true,
		IPortableTensor *	output,
		IPortableTensor *	back_prop_y_pred,
		ir::train::LossReductionType	reduction_type
	)

Definition at line 35 of file LossLayer.cc.

{
  assert(y_pred != nullptr);
  assert(y_true != nullptr);
  assert(output != nullptr);
  // back_prop_y_pred can be nullptr if backwarding is not required
 
  _y_pred = y_pred;
  _y_true = y_true;
  _output = output;
  _back_prop_y_pred = back_prop_y_pred;
  _reduction_type = reduction_type;
}

References _back_prop_y_pred, _output, _reduction_type, _y_pred, and _y_true.

Referenced by onert::backend::train::ops::LossMeanSquaredErrorLayer::configure(), and onert::backend::train::ops::LossCategoricalCrossentropyLayer::configure().

Field Documentation

_back_prop_y_pred

IPortableTensor* onert::backend::train::ops::LossLayer::_back_prop_y_pred

protected

Definition at line 53 of file LossLayer.h.

Referenced by onert::backend::train::ops::LossCategoricalCrossentropyLayer::backward(), onert::backend::train::ops::LossMeanSquaredErrorLayer::backward(), and configure().

_output

IPortableTensor* onert::backend::train::ops::LossLayer::_output

protected

Definition at line 52 of file LossLayer.h.

Referenced by onert::backend::train::ops::LossCategoricalCrossentropyLayer::backward(), configure(), onert::backend::train::ops::LossCategoricalCrossentropyLayer::forward(), and onert::backend::train::ops::LossMeanSquaredErrorLayer::forward().

_reduction_type

ir::train::LossReductionType onert::backend::train::ops::LossLayer::_reduction_type

protected

Definition at line 54 of file LossLayer.h.

Referenced by onert::backend::train::ops::LossCategoricalCrossentropyLayer::backward(), onert::backend::train::ops::LossMeanSquaredErrorLayer::backward(), and configure().

_y_pred

const IPortableTensor* onert::backend::train::ops::LossLayer::_y_pred

protected

Definition at line 50 of file LossLayer.h.

Referenced by onert::backend::train::ops::LossCategoricalCrossentropyLayer::backward(), onert::backend::train::ops::LossMeanSquaredErrorLayer::backward(), configure(), onert::backend::train::ops::LossCategoricalCrossentropyLayer::forward(), and onert::backend::train::ops::LossMeanSquaredErrorLayer::forward().

_y_true

const IPortableTensor* onert::backend::train::ops::LossLayer::_y_true

protected

Definition at line 51 of file LossLayer.h.

Referenced by onert::backend::train::ops::LossCategoricalCrossentropyLayer::backward(), onert::backend::train::ops::LossMeanSquaredErrorLayer::backward(), configure(), onert::backend::train::ops::LossCategoricalCrossentropyLayer::forward(), and onert::backend::train::ops::LossMeanSquaredErrorLayer::forward().

---

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

• runtime/onert/backend/train/ops/LossLayer.h
• runtime/onert/backend/train/ops/LossLayer.cc