skip optimization for dynamic tensors

Samsung · Sep 23, 2024 · c5ddcc5 · c5ddcc5
1 parent 8976ecc
commit c5ddcc5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/runtime/onert/backend/cpu/ops/ExpandDimsLayer.cc b/runtime/onert/backend/cpu/ops/ExpandDimsLayer.cc
@@ -42,7 +42,7 @@ void ExpandDimsLayer::configure(const IPortableTensor *input, IPortableTensor *o
 void ExpandDimsLayer::run()
 {
   bool optimization_applied = false;
-  if ((1 == _input_consumers_number) && !_input->has_padding())
+  if ((1 == _input_consumers_number) && !_input->has_padding() && !_output->is_dynamic())
   {
     optimization_applied = try_assign_buffer(_input, _output);
   }

diff --git a/runtime/onert/backend/cpu/ops/ReshapeLayer.cc b/runtime/onert/backend/cpu/ops/ReshapeLayer.cc
@@ -34,7 +34,7 @@ ReshapeLayer::ReshapeLayer() : _input(nullptr), _shape(nullptr), _output(nullptr
 void ReshapeLayer::reshapeGeneric()
 {
   bool optimization_applied = false;
-  if ((1 == _input_consumers_number) && !_input->has_padding())
+  if ((1 == _input_consumers_number) && !_input->has_padding() && !_output->is_dynamic())
   {
     optimization_applied = try_assign_buffer(_input, _output);
   }