diff options
Diffstat (limited to 'compiler/exo/src/Conversion/DepthwiseConv2DConverter.cpp')
-rw-r--r-- | compiler/exo/src/Conversion/DepthwiseConv2DConverter.cpp | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/compiler/exo/src/Conversion/DepthwiseConv2DConverter.cpp b/compiler/exo/src/Conversion/DepthwiseConv2DConverter.cpp index e3884c3cc..32ad44385 100644 --- a/compiler/exo/src/Conversion/DepthwiseConv2DConverter.cpp +++ b/compiler/exo/src/Conversion/DepthwiseConv2DConverter.cpp @@ -25,6 +25,8 @@ #include <loco/Service/TypeInference.h> #include <loco/Service/ShapeInference.h> +#include <limits> + namespace exo { @@ -75,9 +77,9 @@ bool DepthwiseConv2DConverter::convert(loco::DepthwiseConv2D *origin) reshape->tensor(filter_dec); int32_t new_shape[4] = { - 1, static_cast<int32_t>(filter_shape.height().value()), - static_cast<int32_t>(filter_shape.width().value()), - static_cast<int32_t>(filter_shape.depth().value() * filter_shape.multiplier().value())}; + 1, static_cast<int32_t>(filter_shape.height().value()), + static_cast<int32_t>(filter_shape.width().value()), + static_cast<int32_t>(filter_shape.depth().value() * filter_shape.multiplier().value())}; locoex::set_new_shape(reshape, new_shape, 4); tfl_dw_conv2d->filter(reshape); |