diff options
Diffstat (limited to 'src/armnnTfLiteParser/TfLiteParser.cpp')
-rw-r--r-- | src/armnnTfLiteParser/TfLiteParser.cpp | 14 |
1 files changed, 8 insertions, 6 deletions
diff --git a/src/armnnTfLiteParser/TfLiteParser.cpp b/src/armnnTfLiteParser/TfLiteParser.cpp index fc5041bf9a..a0c673a5ad 100644 --- a/src/armnnTfLiteParser/TfLiteParser.cpp +++ b/src/armnnTfLiteParser/TfLiteParser.cpp @@ -1944,12 +1944,6 @@ void TfLiteParser::ParseReshape(size_t subgraphIndex, size_t operatorIndex) std::vector<int32_t> targetShape; if (inputs.size() > 1 && inputs[1] != nullptr) { - if (options != nullptr) - { - ARMNN_THROW_PARSE_EXCEPTION("Target shape defined in reshape parameters and input tensor. " - "Only one method expected"); - } - if (inputs[1]->is_variable) { ARMNN_THROW_PARSE_EXCEPTION( "Target shapes defined in non-const input tensors is not supported"); @@ -1971,6 +1965,14 @@ void TfLiteParser::ParseReshape(size_t subgraphIndex, size_t operatorIndex) { targetShape.push_back(vals[i]); } + + if (options != nullptr && + options->new_shape.empty() == false && + options->new_shape != targetShape) + { + ARMNN_THROW_PARSE_EXCEPTION("Target shape defined in reshape parameters and as input tensor but " + "the values do not match"); + } } else { |