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

Function UpdateOutputSlotToFp32

src/armnn/NetworkUtils.cpp:19–25  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

17{
18
19void UpdateOutputSlotToFp32(OutputSlot& outputSlot)
20{
21 const TensorInfo& origTensorInfo = outputSlot.GetTensorInfo();
22 TensorInfo newTensorInfo(origTensorInfo);
23 newTensorInfo.SetDataType(DataType::Float32);
24 outputSlot.SetTensorInfo(newTensorInfo);
25}
26
27void ChangeOutputFp16ToFp32(Layer& layer)
28{

Callers 1

ChangeOutputFp16ToFp32Function · 0.85

Calls 2

SetDataTypeMethod · 0.80
SetTensorInfoMethod · 0.45

Tested by

no test coverage detected