昇腾社区首页
中文
注册

函数:set_tensor_const

产品支持情况

产品

是否支持

Atlas A3 训练系列产品/Atlas A3 推理系列产品

Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件

Atlas 训练系列产品

Atlas 推理系列产品

Atlas 200I/500 A2 推理产品

Atlas 200/300/500 推理产品

功能说明

为算子设置constant输入,指定存放输入Tensor数据的内存地址和地址长度。

函数原型

  • C函数原型
    1
    aclError aclSetTensorConst(aclTensorDesc *desc, void *dataBuffer, size_t length)
    
  • python函数
    1
    ret = acl.set_tensor_const(desc, dataBuffer, length)
    

参数说明

参数名

说明

desc

int,aclTensorDesc类型的指针地址。

需提前调用acl.create_tensor_desc接口创建aclTensorDesc类型。

dataBuffer

int,存放constant输入Tensor数据的内存地址。

此处算子输入Tensor数据的内存必须根据应用运行模式来确定。内存申请接口请参见内存管理

  • 应用运行在Host时,此处需申请Host上的内存。
  • 应用运行在Device时,此处需申请Device上的内存。

length

int,内存地址的长度,单位是Byte。

返回值说明

返回值

说明

ret

int,错误码。

  • 返回0表示成功。
  • 返回其它值表示失败。