Ab Android 11 enthält die NNAPI zwei Kontrollflussvorgänge, IF
und WHILE
, die andere Modelle als Argumente annehmen und diese bedingt (IF
) oder wiederholt (WHILE
) ausführen. So lassen sich Modelle erstellen, die je nach Eingabewerten unterschiedliche Vorgänge ausführen oder Vorgänge mehrmals ausführen, ohne sie zu entrollen. Das ist wichtig für Anwendungsfälle wie dynamische RNNs und seq2seq.
In NN HAL 1.3 enthält das Modell mehrere Subgraphen, einschließlich des Haupt-Subgraphs, der zum Angeben von Eingaben und Ausgaben einer Ausführung verwendet wird. Ein untergeordneter Graph kann mithilfe von Operanden vom Typ SUBGRAPH
auf andere untergeordnete Graphen verweisen. Das Framework kann einen Kontrollflussvorgang nur dann an einen Beschleuniger senden, wenn der Beschleuniger alle Vorgänge in allen Untergrafen unterstützt, auf die dieser Kontrollflussvorgang verweist.
HAL-Schnittstellen
In NN HAL 1.3 finden Sie die Definitionen zum Kontrollfluss unter types.hal
.
IF
- undWHILE
-VorgangstypenSUBGRAPH
Operandentyp und die entsprechendeSUBGRAPH
OperandenlebensdauerModel
-Struktur, die den HauptSubgraphen und eine Liste der referenzierten Subgraphen enthältCapabilities
-Struktur, dieifPerformance
undwhilePerformance
enthält
IDevice.hal
enthält IDevice
, dessen Methode getSupportedOperations_1_3()
IF
und WHILE
anders als andere Vorgänge behandeln muss.
IPreparedModel.hal
enthält IPreparedModel
, dessen Methoden execute_1_3()
, executeSynchronously_1_3()
und executeFenced()
ein optionales loopTimeoutDuration
-Argument annehmen.
Treiberimplementierung
Eine Beispielimplementierung für Vorgänge finden Sie unter CpuExecutor::executeIfOperation
und CpuExecutor::executeWhileOperation
.
Beispiele für die Logik zur Validierung von Vorgängen finden Sie unter validateIfOperation()
und validateWhileOperation()
.
Es ist wichtig, arithmetische und Vergleichsoperationen auf TENSOR_INT32
-Operanden der Form 1
zu unterstützen, da diese als Schleifenzähler verwendet werden können.
Ebenso sollten Vorgänge, die TENSOR_BOOL8
-Operanden der Form 1
liefern, mit IF
- und WHILE
-Bedingungen verwendet werden.
Zeitlimit für die Ausführung von WHILE-Schleifen
Um endlose Schleifen zu vermeiden, muss die Ausführung abgebrochen werden, wenn eine WHILE
-Schleife länger dauert als der loopTimeoutDuration
-Wert, der an einen Aufruf von IPreparedModel::execute_1_3()
, IPreparedModel::executeSynchronously_1_3()
oder IPreparedModel::executeFenced()
übergeben wird (oder der Standardwert, falls er weggelassen wird).
Zertifizierungsstufe
Tests der Programmabfolge sind Teil der CTS- und VTS-Testsuites. Weitere Informationen finden Sie unter Validierung.