--reset

--match=.*pool_ci_2d.* # Use 2d problems only from shapes_basic
--mb=2
--tag=axb
--alg=max,avg_np,avg_p

# Training
--dt=f32,bf16,f16
--dir=FWD_D,BWD_D
--batch=shapes_basic

# Inference
--dir=FWD_I
--tag=axb
--dt=f16,s8,u8
--attr-post-ops=,add:f32:per_oc
--batch=shapes_basic
