Feature Scope
Data Types
Propagation Direction |
src Data Type |
dst Data Type |
alg Type |
|---|---|---|---|
forward/backward |
f32 |
f32 |
max, avg_p, avg_np |
forward/backward |
f16 |
f16 |
max, avg_p, avg_np |
forward/backward |
bf16 |
bf16 |
max, avg_p, avg_np |
forward |
s8 |
s8 |
max, avg_p, avg_np |
forward |
u8 |
u8 |
max, avg_p, avg_np |
forward |
s32 |
s32 |
max, avg_p, avg_np |
Data Layout
1D to 5D tensors are supported. The following data layout formats are supported.
Tensor Dimension |
src Data Layout |
dst Data Layout |
|---|---|---|
1D |
a |
a |
2D |
ab,ba |
ab,ba |
3D |
abc,acb,bac,bca,cab,cba |
abc,acb,bac,bca,cab,cba |
4D |
abcd,abdc,acbd,acdb,adbc,adcb,bacd,bcda,cdab,cdba,dcab |
abcd,abdc,acbd,acdb,adbc,adcb,bacd,bcda,cdab,cdba,dcab |
5D |
abcde,abced,abdec,acbde,acdeb,adecb,bacde,bcdea,cdeab,cdeba,decab |
abcde,abced,abdec,acbde,acdeb,adecb,bacde,bcdea,cdeab,cdeba,decab |
Parent topic: Pool