昇腾社区首页
中文
注册

SetLoadDataPaddingValue

功能说明

用于调用Load3Dv1接口/Load3Dv2接口时设置Pad填充的数值。Load3Dv1/Load3Dv2的模板参数isSetPadding设置为true时,用户需要通过本接口设置Pad填充的数值,设置为false时,本接口设置的填充值不生效。

函数原型

1
2
template <typename T>
__aicore__ inline void SetLoadDataPaddingValue(const T padValue)

参数说明

表1 参数说明

参数名称

输入/输出

含义

padValue

输入

Pad填充值的数值。

Atlas 推理系列产品AI Core,支持的数据类型为:int8_t/uint8_t/half/int16_t/uint16_t

Atlas A2 训练系列产品/Atlas 800I A2 推理产品,支持数据类型:int8_t/uint8_t/half/int16_t/uint16_t/bfloat16_t/int32_t/uint32_t/float

Atlas 200I/500 A2 推理产品, 支持数据类型:int8_t/uint8_t/half/int16_t/uint16_t/bfloat16_t/int32_t/uint32_t/float

约束说明

支持的型号

Atlas A2 训练系列产品/Atlas 800I A2 推理产品

Atlas 推理系列产品AI Core

Atlas 200I/500 A2 推理产品

调用示例

参考调用示例