aclmdlGetDatasetBuffer

产品支持情况

  • Ascend 950PR/Ascend 950DT:支持
  • Atlas A3 训练系列产品/Atlas A3 推理系列产品:支持
  • Atlas A2 训练系列产品/Atlas A2 推理系列产品:支持
  • Atlas 200I/500 A2 推理产品:支持
  • Atlas 推理系列产品:支持
  • Atlas 训练系列产品:支持
  • IPV350:支持

功能说明

获取aclmdlDataset中的第n个aclDataBuffer。

函数原型

aclDataBuffer* aclmdlGetDatasetBuffer(const aclmdlDataset *dataset, size_t index)

参数说明

参数名 输入/输出 说明
dataset 输入 aclmdlDataset类型的指针。
需提前调用aclmdlCreateDataset接口创建aclmdlDataset类型的数据。
index 输入 表明获取的是第几个aclDataBuffer。

返回值说明

  • 获取成功,返回aclDataBuffer的地址。
  • 获取失败返回空地址。