| Kevin May | 1bea6be | 2023-12-12 11:18:46 +0000 | [diff] [blame] | 1 | // |
| 2 | // Copyright © 2023 Arm Ltd and Contributors. All rights reserved. | ||||
| 3 | // SPDX-License-Identifier: MIT | ||||
| 4 | // | ||||
| 5 | |||||
| 6 | #pragma once | ||||
| 7 | |||||
| 8 | #include "TosaOperatorUtils.hpp" | ||||
| 9 | |||||
| 10 | using namespace armnn; | ||||
| 11 | using namespace tosa; | ||||
| 12 | |||||
| 13 | TosaSerializationBasicBlock* ConvertSplitToTosaOperator(const Layer* layer, | ||||
| 14 | const std::vector<const TensorInfo*>& inputs, | ||||
| 15 | const std::vector<const TensorInfo*>& outputs, | ||||
| 16 | const SplitterDescriptor* splitDescriptor); | ||||