例
Int dpuGetOutputTensorCnt
(
DPUTask *task,
const char*nodeName
);
引数
- task
- DPU タスクへのポインター。
- nodeName
- DPU ノード名へのポインター。注記: DPU カーネルやタスクの出力ノード名は、ニューラル ネットワークが VAI_C によってコンパイルされた後にリストされます。無効なノード名が指定された場合、エラー メッセージが報告されます。
- idx
- ノードの単一出力テンソルのインデックス。デフォルト値は 0。
説明
1 つの DPU タスクの指定されたノードに対する出力テンソルの総数を取得する。
戻り値
DPU タスクの出力テンソルの総数。
インクルード ファイル
n2cube.h
入力可能なインターフェイス
Vitis AI v1.0