目录

CreateNDTensorBuffer

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

CreateNDTensorBuffer(const NDTensorDesc& tensorDesc)

[h2]接口定义

HIAI_TENSOR_API_EXPORT std::shared_ptr<INDTensorBuffer> CreateNDTensorBuffer(const NDTensorDesc& tensorDesc);

[h2]功能介绍

根据Tensor描述，创建TensorBuffer。

[h2]参数

名称

类型

描述

tensorDesc

const NDTensorDesc&

Tensor描述。

[h2]返回

类型

描述

std::shared_ptr<INDTensorBuffer>

INDTensorBuffer的指针。

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

[h2]接口定义

std::shared_ptr<INDTensorBuffer> CreateNDTensorBuffer(const NDTensorDesc& tensorDesc, void* data, size_t dataSize);

[h2]功能介绍

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

[h2]参数

名称

类型

描述

tensorDesc

const NDTensorDesc&

Tensor描述。

data

void*

Tensor数据地址。

dataSize

size_t

Tensor数据大小。

[h2]返回

类型

描述

std::shared_ptr<INDTensorBuffer>

INDTensorBuffer的指针。

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

[h2]接口定义

std::shared_ptr<INDTensorBuffer> CreateNDTensorBuffer(const NDTensorDesc& tensorDesc, const NativeHandle& handle);

[h2]功能介绍

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

[h2]参数

名称

类型

描述

tensorDesc

const NDTensorDesc&

Tensor描述。

handle

const NativeHandle&

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

[h2]返回

类型

描述

std::shared_ptr<INDTensorBuffer>

INDTensorBuffer的指针。