ONE - On-device Neural Engine
|
Go to the source code of this file.
Namespaces | |
namespace | onert |
namespace | onert::backend |
namespace | onert::backend::train |
namespace | onert::backend::train::ops |
Typedefs | |
using | onert::backend::train::ops::OperandType = onert::ir::DataType |
Functions | |
nnfw::cker::Shape | onert::backend::train::ops::getShape (const IPortableTensor *tensor) |
Get shape of tensor. | |
const IPortableTensor * | onert::backend::train::ops::backpropActivation (const ir::Activation &activation, const IPortableTensor *output, const IPortableTensor *input_backprop, IPortableTensor *output_backprop) |
backpropagate acitvation | |
void | onert::backend::train::ops::biasGrad (const IPortableTensor *input_backprop, IPortableTensor *bias_grad) |
backpropagate bias | |
nnfw::cker::train::LossReductionType | onert::backend::train::ops::convertLossReductionType (ir::train::LossReductionType type) |
convert loss reduction type | |