dropout_tile
-
void ckernel::dropout_kernel_init(uint32_t seed)
-
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
Type
Valid Range
Required
tile_index
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
probability
A non-negative integer value representing dropout probability
uint32_t
0 to INT_MAX (float_probability * (double) INT_MAX)
True
scale_factor
uint bitwise representation of 32 bit floating point scale factor
uint32_t
True