diff options
Diffstat (limited to 'runtime/onert/backend/ruy/BackendContext.h')
-rw-r--r-- | runtime/onert/backend/ruy/BackendContext.h | 22 |
1 files changed, 7 insertions, 15 deletions
diff --git a/runtime/onert/backend/ruy/BackendContext.h b/runtime/onert/backend/ruy/BackendContext.h index b965c9a9d..0dc30f557 100644 --- a/runtime/onert/backend/ruy/BackendContext.h +++ b/runtime/onert/backend/ruy/BackendContext.h @@ -19,7 +19,6 @@ #include <backend/BackendContext.h> #include "TensorBuilder.h" -#include "ConstantInitializer.h" #include "KernelGenerator.h" #include "ExternalContext.h" @@ -33,35 +32,28 @@ namespace ruy class BackendContext : public onert::backend::BackendContext { public: - BackendContext(const Backend *backend, const ir::Graph *graph, + BackendContext(const Backend *backend, ContextData &&data, std::shared_ptr<ITensorRegistry> tensor_registry = nullptr, std::shared_ptr<TensorBuilder> tensor_builder = nullptr, - std::shared_ptr<ConstantInitializer> constant_initializer = nullptr, std::shared_ptr<KernelGenerator> kernel_gen = nullptr) - : onert::backend::BackendContext(backend, graph, tensor_registry), - tensor_builder{tensor_builder}, constant_initializer{constant_initializer}, - kernel_gen{kernel_gen}, _external_context(new ExternalContext) + : onert::backend::BackendContext(backend, std::move(data), tensor_registry), + tensor_builder{tensor_builder}, kernel_gen{kernel_gen}, _external_context(new ExternalContext) { } - ITensorRegistry *genTensors(const std::vector<onert::ir::OpSequenceIndex> &order, - const ir::OpSequences &op_seqs, - const ir::LowerInfoMap &lower_info) override; + ITensorRegistry *genTensors() override; - FunctionMap genKernels(const std::vector<ir::OpSequenceIndex> &order, - const ir::OpSequences &op_seqs) override; + FunctionMap genKernels() override; std::shared_ptr<ExternalContext> external_context() { return _external_context; } private: - void initConsts(); - void planTensors(const std::vector<onert::ir::OpSequenceIndex> &order, - const ir::OpSequences &op_seqs, const ir::LowerInfoMap &lower_info); + void planTensors(const std::vector<onert::ir::OperationIndex> &order, + const compiler::GraphLowerInfo &lower_info); public: // TODO Make it private std::shared_ptr<TensorBuilder> tensor_builder; - std::shared_ptr<ConstantInitializer> constant_initializer; std::shared_ptr<KernelGenerator> kernel_gen; private: |