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