Searched refs:enable_mixed_precision_graph_rewrite_v1 (Results 1 – 3 of 3) sorted by relevance
61 opt = mixed_precision.enable_mixed_precision_graph_rewrite_v1(opt, 123.)72 mixed_precision.enable_mixed_precision_graph_rewrite_v1(opt)82 mixed_precision.enable_mixed_precision_graph_rewrite_v1(opt)100 opt = mixed_precision.enable_mixed_precision_graph_rewrite_v1(opt, 123.)126 opt = mixed_precision.enable_mixed_precision_graph_rewrite_v1(opt, 123.)136 mixed_precision.enable_mixed_precision_graph_rewrite_v1(opt, 123.)181 mixed_precision.enable_mixed_precision_graph_rewrite_v1(194 mixed_precision.enable_mixed_precision_graph_rewrite_v1(
84 def enable_mixed_precision_graph_rewrite_v1(opt, loss_scale='dynamic'): function
33 …sorflow.python.training.experimental.mixed_precision import enable_mixed_precision_graph_rewrite_v1