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);
LayerType GetType() const
std::vector< InputSlot >::iterator BeginInputSlots()
std::vector< InputSlot >::iterator EndInputSlots()
std::vector< ConvertFp32ToFp16Layer * > InsertConvertFp32ToFp16LayersAfter(Graph &graph, Layer &layer)
DataType GetDataType() const
void SetDataType(DataType type)
std::vector< OutputSlot >::iterator EndOutputSlots()
~ConvertFp32NetworkToFp16Impl()=default
DataType GetDataType() const
std::vector< OutputSlot >::iterator BeginOutputSlots()
ConvertFp32NetworkToFp16Impl()=default
std::vector< ConvertFp16ToFp32Layer * > InsertConvertFp16ToFp32LayersBefore(Graph &graph, Layer &layer, bool expectCorrectInputType)
void Run(Graph &graph, Layer &layer) const