diff options
Diffstat (limited to 'src/armnnTfParser/TfParser.cpp')
| -rw-r--r-- | src/armnnTfParser/TfParser.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/armnnTfParser/TfParser.cpp b/src/armnnTfParser/TfParser.cpp index 31f0b42c0d..c00722c4ad 100644 --- a/src/armnnTfParser/TfParser.cpp +++ b/src/armnnTfParser/TfParser.cpp @@ -1305,7 +1305,7 @@ TensorInfo OutputShapeOfExpandDims(const tensorflow::NodeDef& nodeDef, TensorInf // and insert 1 dimension at index 'expandDim' if (expandDim < 0) { - auto outputDimSize = boost::numeric_cast<uint32_t>(outputDims.size() + 1); + int outputDimSize = boost::numeric_cast<int>(outputDims.size() + 1); auto getPosition = std::next(outputDims.begin() + outputDimSize, expandDim); outputDims.insert(getPosition, 1); } |