Feature Scope
Data Types
src Data Type |
dst Data Type |
|---|---|
f32 |
f32 |
f16 |
f16 |
bf16 |
bf16 |
s8 |
s8 |
u8 |
u8 |
The data types of mean, variance, scale, and shift are independent of src and dst, and remains f32.
Data Layout
3D to 5D tensors are supported. The following data layout formats are supported. The layout of input and output tensors must be the same.
Tensor Dimension |
src Data Layout |
dst Data Layout |
|---|---|---|
2D |
ab |
ab |
3D |
abc,acb |
abc,acb |
4D |
abcd,acdb |
abcd,acdb |
5D |
abcde,acdeb |
abcde,acdeb |
Parent topic: Group Normalization