Go to the source code of this file.
|
| | armnn |
| | Copyright (c) 2021 ARM Limited and Contributors.
|
| |
|
| bool | CollapseLeadingUnitDimensions (const TensorInfo &in, TensorInfo &out) |
| |
| template<typename SlotListType > |
| void | BuildAddMulAddSlotLists (bool handleReLu, bool multipleOutputs, std::vector< SlotListType > &inputLayersSlotLists, std::vector< SlotListType > &outputLayersSlotLists) |
| |
| void | GetFusedName (Layer *layerList[4], std::string &fusedName) |
| |
| template<typename Type > |
| bool | BuildAddMulAddTensorInfoLists (Type *layerList[4], unsigned int &numInputs, unsigned int &numOutputs, std::vector< TensorInfo > &inputInfos, std::vector< TensorInfo > &outputInfos, const ActivationDescriptor *&activationDescriptor, bool &fuseReLu) |
| |