diff options
Diffstat (limited to 'src/armnnTestUtils/CommonTestUtils.hpp')
| -rw-r--r-- | src/armnnTestUtils/CommonTestUtils.hpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/src/armnnTestUtils/CommonTestUtils.hpp b/src/armnnTestUtils/CommonTestUtils.hpp index b75a32be61..5b4b356247 100644 --- a/src/armnnTestUtils/CommonTestUtils.hpp +++ b/src/armnnTestUtils/CommonTestUtils.hpp @@ -79,8 +79,11 @@ void SetWeightAndBias(ConvolutionLayer* layer, const armnn::TensorInfo& weightIn layer->m_Weight->Allocate(); layer->m_Bias->Allocate(); } +armnn::SubgraphView::InputSlots CreateInputsFrom(armnn::Layer* layer, + std::vector<unsigned int> ignoreSlots = {}); -armnn::SubgraphView::InputSlots CreateInputsFrom(const std::vector<armnn::Layer*>& layers); +armnn::SubgraphView::InputSlots CreateInputsFrom(const std::vector<armnn::Layer*>& layers, + std::vector<unsigned int> ignoreSlots = {}); armnn::SubgraphView::OutputSlots CreateOutputsFrom(const std::vector<armnn::Layer*>& layers); |