MCPcopy Create free account
hub / github.com/ARM-software/armnn / GetLayerInOutDatatype

Function GetLayerInOutDatatype

src/armnn/Network.cpp:1037–1044  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1035}
1036
1037inline std::vector<DataType> GetLayerInOutDatatype(const Layer* layer)
1038{
1039 DataType dataTypeIn = layer->GetNumInputSlots() == 0 ? DataType::Float32 :
1040 layer->GetInputSlot(0).GetConnectedOutputSlot()->GetTensorInfo().GetDataType();
1041 DataType dataTypeOut = layer->GetNumOutputSlots() == 0 ? DataType::Float32 :
1042 layer->GetOutputSlot(0).GetTensorInfo().GetDataType();
1043 return {dataTypeIn, dataTypeOut};
1044}
1045
1046bool CheckFp16Support(BackendsMap& backends,
1047 const std::vector<BackendId>& availablePreferredBackends)

Callers 2

AssignBackendsFunction · 0.85

Calls 5

GetOutputSlotMethod · 0.80
GetNumInputSlotsMethod · 0.45
GetDataTypeMethod · 0.45
GetNumOutputSlotsMethod · 0.45

Tested by

no test coverage detected