昇腾社区首页
中文
注册
开发者
下载

vlrelu

功能说明

Leaky ReLU 接口,如果元素大于或等于0,则保持原值不变,如果元素小于0,则原值乘以 src1 定义的标量值,计算公式如下:

each_element_of([dst]) = each_element_of([src0]) < 0 ? each_element_of([src0]) * src1 : each_element_of([src0])

以 block(32Byte)为单位完成计算,一次完成8个 block 的计算。

该接口支持通过MASK控制哪些元素参与计算。

接口原型

// 相同接口的不同原型区别在于源地址和目的地址的数据类型不同。
void vlrelu(__ubuf__ half *dst, __ubuf__ half *src0, half src1, uint8_t repeat, uint16_t dstBlockStride, uint16_t src0BlockStride, uint16_t dstRepeatStride, uint16_t src0RepeatStride); 

void vlrelu(__ubuf__ float *dst, __ubuf__ float *src0, float src1, uint8_t repeat, uint16_t dstBlockStride, uint16_t src0BlockStride, uint16_t dstRepeatStride, uint16_t src0RepeatStride);

参数说明

参数含义见 表2 双目运算参数说明

流水类型

PIPE_V