test_conv_bfloat16 524 Bytes
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23
# bfloat16
--reset
--mb=2
--allow-unimpl=true
--dir=FWD_B
--cfg=bf16bf16f32  --batch=conv_resnet_50
--cfg=bf16bf16bf16 --batch=conv_all

--dir=FWD_D
--cfg=bf16bf16bf16 --batch=conv_resnet_50

--dir=BWD_D
--cfg=f32bf16bf16  --batch=conv_resnet_50
--cfg=bf16bf16bf16 --batch=conv_all

--dir=BWD_WB
--cfg=bf16f32bf16  --batch=conv_all
--cfg=bf16bf16bf16 --batch=conv_resnet_50

--dir=FWD_B
--attr=post_ops='sum' --batch=conv_resnet_50
--attr=post_ops='relu' --batch=conv_resnet_50
--attr=post_ops='sum;tanh' --batch=conv_tails