diff options
Diffstat (limited to '1.1/HalPolicy.cpp')
-rw-r--r-- | 1.1/HalPolicy.cpp | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/1.1/HalPolicy.cpp b/1.1/HalPolicy.cpp index 1b1c06ea..d241290f 100644 --- a/1.1/HalPolicy.cpp +++ b/1.1/HalPolicy.cpp @@ -300,19 +300,19 @@ bool HalPolicy::ConvertSqueeze(const Operation& operation, const Model& model, C if(!axisOperand) { axis.assign(dimensionSequence, - dimensionSequence+inputInfo.GetNumDimensions()); + dimensionSequence+rank); } else { GetTensorInt32Values(*axisOperand, axis, model, data); } + std::vector<uint32_t> outputDims; - for (auto& i : axis) + for(unsigned int i = 0; i < rank; i++) { - auto currentDimension = inputInfo.GetShape()[i]; bool skipSqueeze = (std::find(axis.begin(), axis.end(), i) == axis.end()); - + auto currentDimension = inputInfo.GetShape()[i]; if (skipSqueeze || currentDimension != 1) { outputDims.push_back(currentDimension); |