ONE - On-device Neural Engine
|
#include <TrainableBackendContext.h>
Data Fields | |
std::unique_ptr< ir::train::TrainableGraph > | tgraph |
std::vector< onert::ir::OperationIndex > | op_order |
util::Set< ir::OperandIndex > | external_operands |
ir::OperandIndexMap< ir::Layout > | operand_layouts |
std::shared_ptr< custom::IKernelBuilder > | custom_kernel_builder |
bool | is_linear_executor |
ir::train::OptimizerInfo | optim_info |
Definition at line 39 of file TrainableBackendContext.h.
std::shared_ptr<custom::IKernelBuilder> onert::backend::train::TrainableContextData::custom_kernel_builder |
Definition at line 50 of file TrainableBackendContext.h.
util::Set<ir::OperandIndex> onert::backend::train::TrainableContextData::external_operands |
Definition at line 46 of file TrainableBackendContext.h.
bool onert::backend::train::TrainableContextData::is_linear_executor |
Definition at line 52 of file TrainableBackendContext.h.
std::vector<onert::ir::OperationIndex> onert::backend::train::TrainableContextData::op_order |
Definition at line 44 of file TrainableBackendContext.h.
ir::OperandIndexMap<ir::Layout> onert::backend::train::TrainableContextData::operand_layouts |
Definition at line 48 of file TrainableBackendContext.h.
ir::train::OptimizerInfo onert::backend::train::TrainableContextData::optim_info |
Definition at line 54 of file TrainableBackendContext.h.
std::unique_ptr<ir::train::TrainableGraph> onert::backend::train::TrainableContextData::tgraph |
Definition at line 42 of file TrainableBackendContext.h.