• Reed's avatar
    Add --fp16_implementation option. (#6703) · b691578c
    Reed authored
    This options allows the new tf.train.experimental.enable_mixed_precision_graph_rewrite() function to be used for fp16, instead of manual casts.
    b691578c
_performance.py 10.7 KB