rml_op_batch_norm_params
Parameters for the RML_OP_BATCH_NORM operation.
f(x) = scale * (x - mean) / sqrt(variance + epsilon) + bias
API
Graph Manipulation
Items
Name |
Type |
Description |
---|---|---|
input |
Operation that outputs tensor containing input data. |
|
mean |
Constant operation that outputs tensor containing the batch mean. |
|
variance |
Constant operation that outputs tensor containing the batch variance. |
|
scale |
Constant operation that outputs tensor containing the scale. |
|
bias |
Constant operation that outputs tensor containing the batch bias. |
|
epsilon |
float |
Small value to use to avoid division by zero. |