summaryrefslogtreecommitdiff
path: root/runtime/onert/backend/ruy/BackendContext.h
diff options
context:
space:
mode:
Diffstat (limited to 'runtime/onert/backend/ruy/BackendContext.h')
-rw-r--r--runtime/onert/backend/ruy/BackendContext.h22
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: