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

rml_op

Operation that outputs tensor containing input data.

mean

rml_op

Constant operation that outputs tensor containing the batch mean.

variance

rml_op

Constant operation that outputs tensor containing the batch variance.

scale

rml_op

Constant operation that outputs tensor containing the scale.

bias

rml_op

Constant operation that outputs tensor containing the batch bias.

epsilon

float

Small value to use to avoid division by zero.