onert::ir::train::CategoricalCrossentropyParam Struct Reference

#include <LossInfo.h>

Collaboration diagram for onert::ir::train::CategoricalCrossentropyParam:

Data Fields
int32_t	axis
float	label_smoothing

Detailed Description

Definition at line 35 of file LossInfo.h.

Field Documentation

axis

int32_t onert::ir::train::CategoricalCrossentropyParam::axis

Definition at line 37 of file LossInfo.h.

Referenced by onert.experimental.train.metrics.categorical_accuracy.CategoricalAccuracy::__init__(), and onert.experimental.train.metrics.categorical_accuracy.CategoricalAccuracy::update_state().

label_smoothing

float onert::ir::train::CategoricalCrossentropyParam::label_smoothing

Definition at line 38 of file LossInfo.h.

---

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

• runtime/onert/core/include/ir/train/LossInfo.h