commit 7cf0eaa26c1fb29ca9df97e4734ec7c1e10f81c4
author Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> Thu Jan 24 17:05:36 2019 +0000
committer Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> Wed Jan 30 11:25:56 2019 +0000
tree 407b519ede76b235c54907fe80411970741e8a00
parent 28d3d63cc0a33f8396b32fa8347c03912c065911

IVGCVSW-2564 Add support for multiple input and output bindings in InferenceModel

Change-Id: I64d724367d42dca4b768b6c6e42acda714985950

tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp

diff --git a/tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp b/tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp
index 98db023..b752c7c 100644
--- a/tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp
+++ b/tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp
@@ -31,9 +31,9 @@
 
                         typename YoloInferenceModel::Params modelParams;
                         modelParams.m_ModelPath = modelOptions.m_ModelDir + "yolov1_tiny_voc2007_model.caffemodel";
-                        modelParams.m_InputBinding = "data";
-                        modelParams.m_OutputBinding = "fc12";
-                        modelParams.m_InputTensorShape = &inputTensorShape;
+                        modelParams.m_InputBindings = { "data" };
+                        modelParams.m_OutputBindings = { "fc12" };
+                        modelParams.m_InputShapes = { inputTensorShape };
                         modelParams.m_IsModelBinary = true;
                         modelParams.m_ComputeDevice = modelOptions.m_ComputeDevice;
                         modelParams.m_VisualizePostOptimizationModel = modelOptions.m_VisualizePostOptimizationModel;