We should fix the real reason. I make a patch as this was making the CrossentropySoftmaxArgMax1HotWithBias op not being inserted as a stabilization.