Imported Upstream version 1.7.0

[platform/core/ml/nnfw.git] / runtime / onert / core / src / interp / operations / DepthwiseConv2D.cc

diff --git a/runtime/onert/core/src/interp/operations/DepthwiseConv2D.cc b/runtime/onert/core/src/interp/operations/DepthwiseConv2D.cc
index 2682fbd..0473855 100644 (file)
--- a/runtime/onert/core/src/interp/operations/DepthwiseConv2D.cc
+++ b/runtime/onert/core/src/interp/operations/DepthwiseConv2D.cc
@@ -59,10 +59,11 @@ void prepareDepthwiseConv(ExecEnv *env, const ir::Operation &node)
     // Handle unspecified output shape
     const auto &depth_conv_node =
         nnfw::misc::polymorphic_downcast<const ir::operation::DepthwiseConv2D &>(node);
-    const auto infered_output_shapes = shape_inference::inferDepthwiseConv2DShape(
+    const auto infered_output_shape = shape_inference::inferDepthwiseConv2DShape(
         in_tensor->tensorInfo().shape(), kernel_tensor->tensorInfo().shape(),
         depth_conv_node.param());
-    env->allocateIfNeeded(out_index, {infered_output_shapes[0], output_info.typeInfo()});
+    env->allocateIfNeeded(
+        out_index, ir::OperandInfo::createStaticInfo(infered_output_shape, output_info.typeInfo()));
   }
   else
   {