onert_micro::train::losses_functions::MSE Struct Reference

#include <MSE.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 MSE.h.

Member Function Documentation

calculateErrorBackpropagation()

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

static

Definition at line 29 of file MSE.cpp.

{
  for (uint32_t i = 0; i < flat_size; ++i)
  {
    output_grad[i] = (2 * (calculated_data[i] - target_data[i])) / static_cast<float>(flat_size);
  }
}

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/MSE.h
• onert-micro/onert-micro/src/train/losses_functions/MSE.cpp