12 namespace optimizations
48 Layer& base = input->GetConnectedOutputSlot()->GetOwningLayer();
51 TensorInfo convertInfo = input->GetConnection()->GetTensorInfo();
55 input->GetConnection()->SetTensorInfo(convertInfo);
63 TensorInfo convertInfo = output->GetTensorInfo();
67 output->SetTensorInfo(convertInfo);
std::vector< InputSlot >::iterator EndInputSlots()
std::vector< ConvertFp32ToFp16Layer * > InsertConvertFp32ToFp16LayersAfter(Graph &graph, Layer &layer)
std::vector< ConvertFp16ToFp32Layer * > InsertConvertFp16ToFp32LayersBefore(Graph &graph, Layer &layer, bool expectCorrectInputType)
ConvertFp32NetworkToFp16Impl()=default
Copyright (c) 2020 ARM Limited.
std::vector< InputSlot >::iterator BeginInputSlots()
DataType GetDataType() const
void Run(Graph &graph, Layer &layer) const
~ConvertFp32NetworkToFp16Impl()=default
void SetDataType(DataType type)
std::vector< OutputSlot >::iterator BeginOutputSlots()
std::vector< OutputSlot >::iterator EndOutputSlots()
DataType GetDataType() const
LayerType GetType() const