onert::compiler::TensorRegistries Class Reference

#include <TensorRegistries.h>

Public Member Functions
	TensorRegistries ()=default
	TensorRegistries (const onert::backend::BackendContexts &backend_contexts, bool include_builtin)
std::unordered_set< std::shared_ptr< onert::backend::ITensorRegistry > >::const_iterator	begin () const
std::unordered_set< std::shared_ptr< onert::backend::ITensorRegistry > >::const_iterator	end () const
std::shared_ptr< backend::builtin::TensorRegistry >	getBuiltinTensorRegistry () const
backend::ITensor *	getITensor (ir::OperandIndex ind) const
backend::builtin::IOTensor *	getIOTensor (ir::OperandIndex ind) const

Detailed Description

Definition at line 32 of file TensorRegistries.h.

Constructor & Destructor Documentation

TensorRegistries() [1/2]

onert::compiler::TensorRegistries::TensorRegistries ( )

default

TensorRegistries() [2/2]

onert::compiler::TensorRegistries::TensorRegistries ( const onert::backend::BackendContexts &  backend_contexts, bool  include_builtin  )

inline

Definition at line 37 of file TensorRegistries.h.

  {
    for (const auto &[backend, ctx] : backend_contexts)
    {
      auto tensor_reg = ctx->tensor_registry;
      if (backend->config()->id() == backend::builtin::Config::ID)
      {
        _builtin_tensor_reg =
          std::dynamic_pointer_cast<backend::builtin::TensorRegistry>(tensor_reg);
        if (include_builtin)
          _tensor_regs.insert(tensor_reg);
      }
      else
      {
        _tensor_regs.insert(tensor_reg);
      }
    }
  }

References onert::backend::builtin::Config::ID.

Member Function Documentation

begin()

std::unordered_set< std::shared_ptr< onert::backend::ITensorRegistry > >::const_iterator onert::compiler::TensorRegistries::begin ( ) const

inline

Definition at line 56 of file TensorRegistries.h.

  {
    return _tensor_regs.cbegin();
  }

end()

std::unordered_set< std::shared_ptr< onert::backend::ITensorRegistry > >::const_iterator onert::compiler::TensorRegistries::end ( ) const

inline

Definition at line 60 of file TensorRegistries.h.

  {
    return _tensor_regs.cend();
  }

getBuiltinTensorRegistry()

std::shared_ptr< backend::builtin::TensorRegistry > onert::compiler::TensorRegistries::getBuiltinTensorRegistry ( ) const

inline

Definition at line 65 of file TensorRegistries.h.

  {
    return _builtin_tensor_reg;
  }

getIOTensor()

backend::builtin::IOTensor * onert::compiler::TensorRegistries::getIOTensor ( ir::OperandIndex  ind ) const

inline

Definition at line 81 of file TensorRegistries.h.

  {
    return _builtin_tensor_reg->getNativeIOTensor(ind);
  }

Referenced by onert::exec::ExecutorBase::ExecutorBase().

getITensor()

backend::ITensor * onert::compiler::TensorRegistries::getITensor ( ir::OperandIndex  ind ) const

inline

Definition at line 70 of file TensorRegistries.h.

  {
    for (const auto &tensor_reg : _tensor_regs)
    {
      auto tensor = tensor_reg->getITensor(ind);
      if (tensor)
        return tensor;
    }
    return nullptr;
  }

---

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

• runtime/onert/core/src/compiler/TensorRegistries.h