例
DPUTensor * dpuGetInputTotalSize
(
DPUTask *task,
Const char
*tensorName
)
引数
- task
- DPU タスクへのポインター。
- tensorName
- モデルがコンパイルされた後に VAI_C コンパイラによって一覧表示されるテンソル名。
説明
指定したテンソル名から DPU タスクの境界入力または出力テンソルを取得します。モデルがコンパイルされた後に VAI_C コンパイラによってテンソル名が一覧表示されます。
戻り値
DPUTensor へのポインター。
インクルード ファイル
n2cube.h
入力可能なインターフェイス
Vitis AI v1.0