selu_tile

void ckernel::selu_tile_init()

Please refer to documentation for any_init.

template<bool is_fp32_dest_acc_en = DST_ACCUM_MODE>
void ckernel::selu_tile(uint32_t idst, uint32_t scale, uint32_t alpha)

Performs element-wise computation of selu = scale * (max(0,x) + min(0, alpha * (exp(x)-1))), where x is each element of a tile in DST register at index tile_index. scale and alpha are each passed as the raw bits of a float. The DST register buffer must be in acquired state via acquire_dst call. This call is blocking and is only available on the compute engine.

Return value: None

Argument Description

idst The index of the tile in DST register buffer to perform the computation on

scale Scale used in selu calculation, as the raw bits of a float

alpha Alpha used in selu calculation, as the raw bits of a float