dropout_tile

void ckernel::dropout_kernel_init(uint32_t seed = 0)

This init should be called once in kernel

void ckernel::dropout_tile(uint32_t idst, uint32_t probability, uint32_t scale_factor)

Performs element-wise dropout on each element of a of a tile in DST register at index tile_index. That is each element may be zeroed out based on a given probability or it may be scaled by a given scale factor. 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

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

probability A non-negative integer value representing dropout probability

scale_factor uint bitwise representation of 32 bit floating point scale factor