onert_micro::train::KernelBuiltinTrainRegistry Class Reference

#include <OMBackpropExecutionBuilder.h>

Public Member Functions
constexpr	KernelBuiltinTrainRegistry ()
OMStatus	getKernelTrainFunc (core::OMBuilderID builderID, KernelTrainFunc **train_func) const

Detailed Description

Definition at line 42 of file OMBackpropExecutionBuilder.h.

Constructor & Destructor Documentation

KernelBuiltinTrainRegistry()

constexpr onert_micro::train::KernelBuiltinTrainRegistry::KernelBuiltinTrainRegistry ( )

inlineconstexpr

Definition at line 45 of file OMBackpropExecutionBuilder.h.

                                         : _operator_train()
  {
#define REGISTER_TRAIN_KERNEL(builtin_operator, name)                        \
  registerKernelTrain(core::OMBuilderID::BuiltinOperator_##builtin_operator, \
                      train_kernel_Circle##name);
 
#include "KernelsToTrain.lst"
 
#undef REGISTER_TRAIN_KERNEL
  }

Member Function Documentation

getKernelTrainFunc()

OMStatus onert_micro::train::KernelBuiltinTrainRegistry::getKernelTrainFunc ( core::OMBuilderID  builderID, KernelTrainFunc **  train_func  ) const

inline

Definition at line 57 of file OMBackpropExecutionBuilder.h.

  {
    const auto builder_id_opcode = size_t(builderID);
    assert(builder_id_opcode < size_t(core::OMBuilderID::BuiltinOperatorsSize));
    if (builder_id_opcode >= size_t(core::OMBuilderID::BuiltinOperatorsSize))
    {
      *train_func = nullptr;
      return UnknownError;
    }
    *train_func = _operator_train[builder_id_opcode];
    return Ok;
  }

References onert_micro::core::BuiltinOperatorsSize, onert_micro::Ok, and onert_micro::UnknownError.

Referenced by onert_micro::train::OMBackpropExecute::runBackward().

---

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

• onert-micro/onert-micro/include/train/OMBackpropExecutionBuilder.h