Visible to Intel only — GUID: GUID-7A31C619-516A-4D77-A64A-BC1BFF91F136
Visible to Intel only — GUID: GUID-7A31C619-516A-4D77-A64A-BC1BFF91F136
BatchNormTrainingBackward
General
BatchNormTrainingBackward operation calculated the gradients of input tensors.
Operation attributes
Attribute Name |
Description |
Value Type |
Supported Values |
Required or Optional |
---|---|---|---|---|
A number to be added to the variance to avoid division by zero. |
f32 |
A positive f32 value |
Required |
|
Controls how to interpret the shape of src and dst . |
string |
NCX , NXC (default) |
Optional |
Execution arguments
The inputs and outputs must be provided according to below index order when constructing an operation.
Inputs
Index |
Argument Name |
Required or Optional |
---|---|---|
0 |
src |
Required |
1 |
diff_dst |
Required |
2 |
mean |
Required |
3 |
variance |
Required |
4 |
gamma ( ) |
Optional |
Outputs
Index |
Argument Name |
Required or Optional |
---|---|---|
0 |
diff_src |
Required |
1 |
diff_gamma |
Optional |
2 |
diff_beta |
Optional |
Supported data types
BatchNormTrainingBackward operation supports the following data type combinations.
Src / Diff_dst / Dst |
Mean / Variance / Gamma / Diff_gamma / Diff_beta |
---|---|
f32 |
f32 |
bf16 |
f32, bf16 |
f16 |
f32 |