relu_tile
-
void ckernel::relu_tile_init()
-
Please refer to documentation for any_init.
-
void ckernel::relu_tile(uint32_t idst)
-
Performs element-wise computation of relu (0 if negative else 1) on each element of a tile in DST register at index tile_index. 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 |
relu_max_tile
-
void ckernel::relu_max_tile_init()
-
Please refer to documentation for any_init.
-
void ckernel::relu_max_tile(uint32_t idst, uint32_t param0)
-
Performs element-wise computation of relu max (relu(max(x, upper_limit))) on each element of a tile in DST register at index tile_index. 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 | | upper_limit | Upper limit of relu_min | uint32_t | Greater than 0 | True |
relu_min_tile
-
void ckernel::relu_min_tile_init()
-
Please refer to documentation for any_init.
-
void ckernel::relu_min_tile(uint32_t idst, uint32_t param0)
-
Performs element-wise computation of relu min (relu(min(x, lower_limit))) on each element of a tile in DST register at index tile_index. 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 | | lower_limit | Upper limit of relu_min | uint32_t | Greater than 0 | True |
leaky_relu_tile
-
void ckernel::leaky_relu_tile_init()
-
Please refer to documentation for any_init.
-
void ckernel::leaky_relu_tile(uint32_t idst, uint32_t slope)
-
Performs element-wise computation of leaky relu (relu(x) + slope*-relu(-x)) on each element of a tile in DST register at index tile_index. 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 | | slope | slope used in leaky relu - will reinterpret unsigned int to float | uint32_t | Greater than 0 | True |