diff options
author | Ferran Balaguer <ferran.balaguer@arm.com> | 2019-02-08 17:09:55 +0000 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-02-11 08:48:53 +0000 |
commit | c602f29d57f34b6bf0805d379b2174667d8bf52f (patch) | |
tree | 50cdc475ec8732575c0cf2c56d4ced770215c4a2 /src/armnnTfParser/TfParser.cpp | |
parent | 9c5d33a26ebc4be391ae4da9de584be2e453c78f (diff) | |
download | armnn-c602f29d57f34b6bf0805d379b2174667d8bf52f.tar.gz |
IVGCVSW-2529 DeepSpeech v1 test
Change-Id: Ieb99ac1aa347cee4b28b831753855c4614220648
Diffstat (limited to 'src/armnnTfParser/TfParser.cpp')
-rwxr-xr-x | src/armnnTfParser/TfParser.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/armnnTfParser/TfParser.cpp b/src/armnnTfParser/TfParser.cpp index 80962748ce..b5fe6be075 100755 --- a/src/armnnTfParser/TfParser.cpp +++ b/src/armnnTfParser/TfParser.cpp @@ -3160,12 +3160,12 @@ void TfParser::LoadNodeDef(const tensorflow::NodeDef& nodeDef, const tensorflow: type = attr.type(); } - if (type != tensorflow::DT_FLOAT && nodeDef.op() != "Const") + if ((type != tensorflow::DT_FLOAT && type != tensorflow::DT_INT32) && nodeDef.op() != "Const") { throw ParseException( boost::str( boost::format( - "Currently only FLOAT is supported for tensorflow nodes (apart from Const). " + "Currently only FLOAT and INT32 are supported for tensorflow nodes (apart from Const). " "Got %1% for Node %2% %3%") % tensorflow::DataType_Name(type) % nodeDef.name() |