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