diff options
author | Finn Williams <Finn.Williams@arm.com> | 2020-07-03 10:12:03 +0100 |
---|---|---|
committer | Finn Williams <Finn.Williams@arm.com> | 2020-07-10 19:27:07 +0100 |
commit | 87d0bda9b49d9df4455f1887027e5ead2527c27e (patch) | |
tree | c58787cce03027d3e1969a169f162f59a8b06f37 /src/armnn/test/FlowControl.cpp | |
parent | c9f74d775da0039fd899f9ee6ec02b98ad575250 (diff) | |
download | armnn-87d0bda9b49d9df4455f1887027e5ead2527c27e.tar.gz |
IVGCVSW-4929 Implement ShapeInferenceMethod in all Layers
Signed-off-by: Finn Williams <Finn.Williams@arm.com>
Change-Id: I2c2d99f97cf89814140b057a9f93f41b364197f5
Diffstat (limited to 'src/armnn/test/FlowControl.cpp')
-rw-r--r-- | src/armnn/test/FlowControl.cpp | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/src/armnn/test/FlowControl.cpp b/src/armnn/test/FlowControl.cpp index 3bc993b33b..6198ca8a24 100644 --- a/src/armnn/test/FlowControl.cpp +++ b/src/armnn/test/FlowControl.cpp @@ -28,13 +28,18 @@ BOOST_AUTO_TEST_CASE(ErrorOnLoadNetwork) std::vector<uint8_t> falseData = {0}; ConstTensor falseTensor(armnn::TensorInfo({1}, armnn::DataType::Boolean), falseData); - IConnectableLayer* constLayer = net->AddConstantLayer(falseTensor, "const"); + IConnectableLayer* constLayer = net->AddConstantLayer(falseTensor, "const"); constLayer->GetOutputSlot(0).SetTensorInfo(armnn::TensorInfo({1}, armnn::DataType::Boolean)); IConnectableLayer* input = net->AddInputLayer(0); + input->GetOutputSlot(0).SetTensorInfo(armnn::TensorInfo({1}, armnn::DataType::Boolean)); IConnectableLayer* switchLayer = net->AddSwitchLayer("switch"); + switchLayer->GetOutputSlot(0).SetTensorInfo(armnn::TensorInfo({1}, armnn::DataType::Boolean)); + switchLayer->GetOutputSlot(1).SetTensorInfo(armnn::TensorInfo({1}, armnn::DataType::Boolean)); + IConnectableLayer* mergeLayer = net->AddMergeLayer("merge"); + mergeLayer->GetOutputSlot(0).SetTensorInfo(armnn::TensorInfo({1}, armnn::DataType::Boolean)); IConnectableLayer* output = net->AddOutputLayer(0); |