22.11
|
#include "TosaMappings.hpp"
Go to the source code of this file.
Functions | |
void | SetBasicBlockConstantTensorData (Layer *layer, TosaSerializationBasicBlock *) |
TosaSerializationBasicBlock * | GetTosaMapping (const LayerType type, const std::vector< const TensorInfo *> &inputs, const std::vector< const TensorInfo *> &outputs, const BaseDescriptor &, bool isMain=false) |
TosaSerializationBasicBlock * | GetTosaMappingFromLayer (Layer *layer, bool isMain=false) |
TosaSerializationBasicBlock* GetTosaMapping | ( | const LayerType | type, |
const std::vector< const TensorInfo *> & | inputs, | ||
const std::vector< const TensorInfo *> & | outputs, | ||
const BaseDescriptor & | , | ||
bool | isMain = false |
||
) |
Definition at line 26 of file TosaMappings.cpp.
References armnn::Addition, and ConvertAdditionToTosaOperator().
Referenced by GetTosaMappingFromLayer(), and TosaRefLayerSupport::IsLayerSupported().
TosaSerializationBasicBlock* GetTosaMappingFromLayer | ( | Layer * | layer, |
bool | isMain = false |
||
) |
Definition at line 47 of file TosaMappings.cpp.
References Layer::GetInputSlots(), Layer::GetOutputSlots(), Layer::GetParameters(), GetTosaMapping(), Layer::GetType(), and SetBasicBlockConstantTensorData().
Referenced by TosaRefBackend::OptimizeSubgraphView().
void SetBasicBlockConstantTensorData | ( | Layer * | layer, |
TosaSerializationBasicBlock * | |||
) |
Definition at line 11 of file TosaMappings.cpp.
References armnn::Convolution2d, and Layer::GetType().
Referenced by GetTosaMappingFromLayer().