ONE - On-device Neural Engine
Loading...
Searching...
No Matches
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 31 of file BackendResolver.h.

Member Function Documentation

◆ getBackend()

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

Definition at line 34 of file BackendResolver.h.

35 {
36 return _gen_map.at(index);
37 }

◆ iterate()

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

Definition at line 45 of file BackendResolver.h.

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

◆ setBackend()

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

Definition at line 39 of file BackendResolver.h.

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

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