diff --git a/runtime/onert/backend/trix/Backend.h b/runtime/onert/backend/trix/Backend.h index a63839720e8..1ad01743d2d 100644 --- a/runtime/onert/backend/trix/Backend.h +++ b/runtime/onert/backend/trix/Backend.h @@ -44,7 +44,7 @@ class Backend : public ::onert::backend::Backend auto &graph = *data.graph; auto context = std::make_unique(this, std::move(data)); auto tr = std::make_shared(); - auto tb = std::make_shared(tr); + auto tb = std::make_shared(tr, context->data().shared_memory_operand_map); context->tensor_registry = tr; context->tensor_builder = tb; context->kernel_gen = std::make_shared(graph, tb, tr, context->dev_context()); diff --git a/runtime/onert/backend/xnnpack/Backend.h b/runtime/onert/backend/xnnpack/Backend.h index 67494a534cd..593f2ae075e 100644 --- a/runtime/onert/backend/xnnpack/Backend.h +++ b/runtime/onert/backend/xnnpack/Backend.h @@ -45,7 +45,7 @@ class Backend : public ::onert::backend::Backend auto &graph = *data.graph; auto context = std::make_unique(this, std::move(data)); auto tr = std::make_shared(); - auto tb = std::make_shared(tr); + auto tb = std::make_shared(tr, context->data().shared_memory_operand_map); context->tensor_registry = tr; context->tensor_builder = tb; context->kernel_gen = std::make_shared(graph, tb, tr, custom_kernel_builder,