Fix all deprecation warnings.

1. Update README.md
2. argmax, use axis instead of deminsion
3. use tf.profiler.profile instead of model_analyzer.print_model_analysis