Imported Upstream version 1.25.0

[platform/core/ml/nnfw.git] / onert-micro / luci-interpreter / src / kernels / ExpandDims.cpp

diff --git a/onert-micro/luci-interpreter/src/kernels/ExpandDims.cpp b/onert-micro/luci-interpreter/src/kernels/ExpandDims.cpp
index 867af01..1035bb8 100644 (file)
--- a/onert-micro/luci-interpreter/src/kernels/ExpandDims.cpp
+++ b/onert-micro/luci-interpreter/src/kernels/ExpandDims.cpp
@@ -1,6 +1,5 @@
 /*
  * Copyright (c) 2022 Samsung Electronics Co., Ltd. All Rights Reserved
- * Copyright 2019 The TensorFlow Authors. All Rights Reserved.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -41,6 +40,7 @@ void configure_kernel_CircleExpandDims(const circle::Operator *cur_op,
   assert(output != nullptr);
 
   auto axis_data = runtime_graph->getConstDataByTensor(axis);
+  assert(axis_data != nullptr);
 
   int32_t axis_value;
 
@@ -65,7 +65,7 @@ void configure_kernel_CircleExpandDims(const circle::Operator *cur_op,
 }
 
 void execute_kernel_CircleExpandDims(const circle::Operator *cur_op,
-                                     BaseRuntimeGraph *runtime_graph, bool is_inplace)
+                                     BaseRuntimeGraph *runtime_graph)
 {
   const auto input_index = cur_op->inputs()->operator[](0);
   const auto output_index = cur_op->outputs()->operator[](0);
@@ -76,6 +76,8 @@ void execute_kernel_CircleExpandDims(const circle::Operator *cur_op,
   const auto input = runtime_graph->getCircleTensorByIndex(input_index);
   const auto output = runtime_graph->getCircleTensorByIndex(output_index);
 
+  bool is_inplace = runtime_graph->is_inplace_op(cur_op);
+
   if (is_inplace)
   {
     runtime_graph->makeInplaceOperation(input, output);