diff options
Diffstat (limited to 'src/armnnTfLiteParser/TfLiteParser.cpp')
| -rw-r--r-- | src/armnnTfLiteParser/TfLiteParser.cpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/armnnTfLiteParser/TfLiteParser.cpp b/src/armnnTfLiteParser/TfLiteParser.cpp index d3eed9cfb1..10bb0f6e97 100644 --- a/src/armnnTfLiteParser/TfLiteParser.cpp +++ b/src/armnnTfLiteParser/TfLiteParser.cpp @@ -2004,6 +2004,11 @@ void TfLiteParser::ParseResize(size_t subgraphIndex, size_t operatorIndex, Resiz case ResizeMethod::Bilinear: { layerName += str(boost::format("BILINEAR:%1%:%2%") % subgraphIndex % operatorIndex); + + const auto & operatorPtr = m_Model->subgraphs[subgraphIndex]->operators[operatorIndex]; + const auto * options = operatorPtr->builtin_options.AsResizeBilinearOptions(); + + desc.m_BilinearAlignCorners = options->align_corners; break; } case ResizeMethod::NearestNeighbor: |