diff options
Diffstat (limited to 'runtime/onert/backend/ruy/KernelGenerator.cc')
-rw-r--r-- | runtime/onert/backend/ruy/KernelGenerator.cc | 10 |
1 files changed, 3 insertions, 7 deletions
diff --git a/runtime/onert/backend/ruy/KernelGenerator.cc b/runtime/onert/backend/ruy/KernelGenerator.cc index c2f6a1f79..b2bbf9bfc 100644 --- a/runtime/onert/backend/ruy/KernelGenerator.cc +++ b/runtime/onert/backend/ruy/KernelGenerator.cc @@ -42,17 +42,13 @@ std::unique_ptr<exec::FunctionSequence> KernelGenerator::generate(ir::OperationI assert(_tensor_builder->dynamicTensorManager()); assert(_tensor_reg); - auto dyn_shape_inferer = std::make_shared<exec::DynamicShapeInferer>(_ctx, _tensor_reg); - // Prepare to handle dynamic tensors later auto dyn_ctx = std::make_shared<exec::FunctionSequence::DynamicTensorCtx>(); { - dyn_ctx->op_ind = ind; - dyn_ctx->operations = &_operations_ctx; - dyn_ctx->dynamic_shape_inferer = std::move(dyn_shape_inferer); - - ret->dynamic_tensor_ctx(dyn_ctx); + dyn_ctx->op = &_operations_ctx.at(ind); + dyn_ctx->dynamic_shape_inferer = std::make_shared<exec::DynamicShapeInferer>(_ctx, _tensor_reg); } + ret->dynamic_tensor_ctx(dyn_ctx); auto &op = _graph.operations().at(ind); op.accept(*this); |