Closed deven-amd closed 4 years ago
This PR/commit adds a command line option auto_mixed_precision to enable AMP (automatic mixed precision) for the tf_cnn_benchmarks. Based on the information given here : https://www.tensorflow.org/api_docs/python/tf/train/experimental/enable_mixed_precision_graph_rewrite
auto_mixed_precision
closing PR
This PR/commit adds a command line option
auto_mixed_precision
to enable AMP (automatic mixed precision) for the tf_cnn_benchmarks. Based on the information given here : https://www.tensorflow.org/api_docs/python/tf/train/experimental/enable_mixed_precision_graph_rewrite