onert_micro::train::losses_functions::SparseCrossEntropy Struct Reference

#include <SparseCrossEntropy.h>

Static Public Member Functions
static void	calculateErrorBackpropagation (const uint32_t flat_size, const float *calculated_data, const float *target_data, float *output_grad)

Detailed Description

Definition at line 32 of file SparseCrossEntropy.h.

Member Function Documentation

calculateErrorBackpropagation()

void SparseCrossEntropy::calculateErrorBackpropagation ( const uint32_t  flat_size, const float *  calculated_data, const float *  target_data, float *  output_grad  )

static

Definition at line 33 of file SparseCrossEntropy.cpp.

{
  uint32_t label = static_cast<uint32_t>(target_data[0]);
 
  for (uint32_t i = 0; i < flat_size; ++i)
  {
    if (i == label)
    {
      output_grad[i] = -1.0 / (calculated_data[i] + float(10.0e-20));
    }
    else
    {
      output_grad[i] = 0.0;
    }
  }
}

Referenced by onert_micro::core::train::OMTrainingHandler::handleError().

---

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

• onert-micro/onert-micro/include/train/losses_functions/SparseCrossEntropy.h
• onert-micro/onert-micro/src/train/losses_functions/SparseCrossEntropy.cpp