diff options
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); |