From 09d711c382e99705ca67968e743bcf2f8be711d3 Mon Sep 17 00:00:00 2001 From: Hanjoung Lee Date: Thu, 20 Aug 2020 17:10:41 +0900 Subject: [PATCH] [onert] cpu KernelGen use _tensor_reg directly (#3895) cpu KernelGen use _tensor_reg directly ONE-DCO-1.0-Signed-off-by: Hanjoung Lee --- runtime/onert/backend/cpu/KernelGenerator.cc | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/runtime/onert/backend/cpu/KernelGenerator.cc b/runtime/onert/backend/cpu/KernelGenerator.cc index c7088e1..a8d829e 100644 --- a/runtime/onert/backend/cpu/KernelGenerator.cc +++ b/runtime/onert/backend/cpu/KernelGenerator.cc @@ -182,11 +182,11 @@ void KernelGenerator::visit(const ir::OpSequence &op_seq) { assert(!_return_fn_seq); assert(_tensor_builder->dynamicTensorManager()); - assert(_tensor_builder->tensorRegistry()); + assert(_tensor_reg); auto dyn_tensor_manager = _tensor_builder->dynamicTensorManager(); - auto dyn_shape_inferer = std::make_shared( - _ctx, dyn_tensor_manager, _tensor_builder->tensorRegistry()); + auto dyn_shape_inferer = + std::make_shared(_ctx, dyn_tensor_manager, _tensor_reg); _return_fn_seq = std::make_unique(); @@ -196,7 +196,7 @@ void KernelGenerator::visit(const ir::OpSequence &op_seq) dyn_ctx->op_seq = &op_seq; dyn_ctx->operations = &_operations_ctx; dyn_ctx->dynamic_shape_inferer = std::move(dyn_shape_inferer); - dyn_ctx->tensor_registry = _tensor_builder->tensorRegistry(); + dyn_ctx->tensor_registry = _tensor_reg; dyn_ctx->dynamic_tensor_manager = _tensor_builder->dynamicTensorManager(); _return_fn_seq->dynamic_tensor_ctx(dyn_ctx); -- 2.7.4