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