Additional fixing following last commit for using single d0 and d1 bias
Mha train develop reduce interface