diff options
author | Kevin May <kevin.may@arm.com> | 2021-04-28 16:16:22 +0100 |
---|---|---|
committer | Kevin May <kevin.may@arm.com> | 2021-04-29 12:38:01 +0000 |
commit | eb03e0fa0f6e8d133efe7d54831ad70da9431874 (patch) | |
tree | 1cc0b63450ba58a570974c7d3feba7b53cf3f8eb /tests/ImageTensorGenerator/ImageTensorGenerator.cpp | |
parent | a04a9d7c11f28c7e932435535e80223782f369f2 (diff) | |
download | armnn-eb03e0fa0f6e8d133efe7d54831ad70da9431874.tar.gz |
IVGCVSW-5744 Remove Tensorflow, Caffe and Quantizer from documentation
* Remove from .md files and Doxygen
* Remove from armnn/docker build
* Remove Tensorflow model format from ExecuteNetworkParams
* Remove Tensorflow model format from ImageTensorGenerator
Signed-off-by: Kevin May <kevin.may@arm.com>
Change-Id: Id6ed4a7d90366c396e8e0395d0ce43a3bcddcee6
Diffstat (limited to 'tests/ImageTensorGenerator/ImageTensorGenerator.cpp')
-rw-r--r-- | tests/ImageTensorGenerator/ImageTensorGenerator.cpp | 8 |
1 files changed, 2 insertions, 6 deletions
diff --git a/tests/ImageTensorGenerator/ImageTensorGenerator.cpp b/tests/ImageTensorGenerator/ImageTensorGenerator.cpp index 34dbe1e352..5417eb4055 100644 --- a/tests/ImageTensorGenerator/ImageTensorGenerator.cpp +++ b/tests/ImageTensorGenerator/ImageTensorGenerator.cpp @@ -164,7 +164,7 @@ public: ("f,model-format", "Format of the intended model file that uses the images." "Different formats have different image normalization styles." - "Accepted values (tensorflow, tflite)", + "Accepted value (tflite)", cxxopts::value<std::string>(m_ModelFormat)) ("o,outfile", "Output raw tensor file path", @@ -235,11 +235,7 @@ public: unsigned int GetNewHeight() {return static_cast<unsigned int>(std::stoi(m_NewHeight));} SupportedFrontend GetModelFormat() { - if (m_ModelFormat == "tensorflow") - { - return SupportedFrontend::TensorFlow; - } - else if (m_ModelFormat == "tflite") + if (m_ModelFormat == "tflite") { return SupportedFrontend::TFLite; } |