ONE - On-device Neural Engine
All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Modules Pages
onert::compiler::BackendResolver Class Reference

#include <BackendResolver.h>

Public Member Functions

const backend::BackendgetBackend (const ir::OperationIndex &index) const
 
void setBackend (const ir::OperationIndex &index, const backend::Backend *backend)
 
void iterate (const std::function< void(const ir::OperationIndex &, const backend::Backend &)> &fn) const
 

Detailed Description

Definition at line 29 of file BackendResolver.h.

Member Function Documentation

◆ getBackend()

const backend::Backend * onert::compiler::BackendResolver::getBackend ( const ir::OperationIndex index) const
inline

Definition at line 32 of file BackendResolver.h.

33 {
34 return _gen_map.at(index);
35 }

◆ iterate()

void onert::compiler::BackendResolver::iterate ( const std::function< void(const ir::OperationIndex &, const backend::Backend &)> &  fn) const
inline

Definition at line 43 of file BackendResolver.h.

44 {
45 for (const auto &[op_index, backend] : _gen_map)
46 {
47 fn(op_index, *backend);
48 }
49 }

◆ setBackend()

void onert::compiler::BackendResolver::setBackend ( const ir::OperationIndex index,
const backend::Backend backend 
)
inline

Definition at line 37 of file BackendResolver.h.

38 {
39 _gen_map[index] = backend;
40 }
loco::GraphInputIndex index(const TFPlaceholder *node)
Definition TFNode.cpp:54

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