Add a command line option to enable automatic mixed precision.

tensorflow / benchmarks

A benchmark framework for Tensorflow

Apache License 2.0

1.15k stars 632 forks source link

Closed deven-amd closed 4 years ago

deven-amd commented 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

deven-amd commented 4 years ago

closing PR