silu_tile
-
void ckernel::silu_tile_init()
-
void ckernel::silu_tile(uint32_t idst)
-
Performs SILU (same as Swish) operation on each element of a tile in DST register at index tile_index. Uses the following implementation: Silu[x] = x*Sigmoid[x]
Ref: https://pytorch.org/docs/stable/generated/torch.nn.SiLU.html?highlight=silu#torch.nn.SiLU
Return value: None
Argument
Description
Type
Valid Range
Required
idst
The index of the tile in DST register buffer to perform the computation on
uint32_t
Must be less than the size of the DST register buffer
True