"docs/vscode:/vscode.git/clone" did not exist on "832b84525727f6fc81230c59ecec204adafac056"
Commit 9fe85228 authored by Reed Wanderman-Milne's avatar Reed Wanderman-Milne Committed by A. Unique TensorFlower
Browse files

Pass use_experimental_api=True to mixed precision functions.

This has no functional impact since the default is currently True, but I plan on changing the default to False soon.

PiperOrigin-RevId: 367049942
parent 5cdbcac3
...@@ -17,6 +17,10 @@ ...@@ -17,6 +17,10 @@
import tensorflow as tf import tensorflow as tf
# TODO(b/181616568): Switch default value of `use_experimental_api` in both
# of these functions to False.
def configure_optimizer(optimizer, def configure_optimizer(optimizer,
use_float16=False, use_float16=False,
use_graph_rewrite=False, use_graph_rewrite=False,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment