文档管理中心
API参考CANN模型管家V2接口模型Tensor创建类INDTensorBufferCreateNDTensorBuffer

CreateNDTensorBuffer

说明

此接口从DDK 5.0.1.0版本起废弃,请使用模型管家V1接口Init(const TensorDimension* dim, HIAI_DataType pdataType)替代。

CreateNDTensorBuffer(const NDTensorDesc& tensorDesc)

接口定义

收起
自动换行
深色代码主题
复制
  1. HIAI_TENSOR_API_EXPORT std::shared_ptr<INDTensorBuffer> CreateNDTensorBuffer(const NDTensorDesc& tensorDesc);

功能介绍

根据Tensor描述,创建TensorBuffer。

参数

展开

名称

类型

描述

tensorDesc

const NDTensorDesc&

Tensor描述。

返回

展开

类型

描述

std::shared_ptr<INDTensorBuffer>

INDTensorBuffer的指针。

CreateNDTensorBuffer(const NDTensorDesc& tensorDesc, void* data, size_t dataSize)

接口定义

收起
自动换行
深色代码主题
复制
  1. std::shared_ptr<INDTensorBuffer> CreateNDTensorBuffer(const NDTensorDesc& tensorDesc, void* data, size_t dataSize);

功能介绍

根据Tensor描述,创建TensorBuffer,同时写入tensor数据和大小。

参数

展开

名称

类型

描述

tensorDesc

const NDTensorDesc&

Tensor描述。

data

void*

Tensor数据地址。

dataSize

size_t

Tensor数据大小。

返回

展开

类型

描述

std::shared_ptr<INDTensorBuffer>

INDTensorBuffer的指针。

CreateNDTensorBuffer(const NDTensorDesc& tensorDesc, const NativeHandle& handle)

接口定义

收起
自动换行
深色代码主题
复制
  1. std::shared_ptr<INDTensorBuffer> CreateNDTensorBuffer(const NDTensorDesc& tensorDesc, const NativeHandle& handle);

功能介绍

根据Tensor描述和NativeHandle,创建TensorBuffer。

参数

展开

名称

类型

描述

tensorDesc

const NDTensorDesc&

Tensor描述。

handle

const NativeHandle&

输入NativeHandle的结构信息,NativeHandle只支持ION内存。

返回

展开

类型

描述

std::shared_ptr<INDTensorBuffer>

INDTensorBuffer的指针。

在 API参考 中进行搜索
请输入您想要搜索的关键词