昇腾社区首页
中文
注册

vlrelu

功能说明

Leaky ReLU运算:

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

函数原型

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);

流水类型

PIPE_V