Learning Discretized Neural Networks under Ricci Flow