experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Remove experimental_export_device_assignment from TPUEstimator.export_savedmodel(), so as to remove rewrite_for_inference().

Browse Source 
As a replacement, export_savedmodel() V2 API supports device_assignment where user call tpu.rewrite in model_fn and pass in device_assigment there.

PiperOrigin-RevId: 251950039
This commit is contained in:
A. Unique TensorFlower 2019-06-06 16:22:15 -07:00 committed by TensorFlower Gardener
parent 3925711f6a
commit 842abcfab5
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/tools/api/golden/v1/tensorflow.estimator.tpu.-t-p-u-estimator.pbtxt
View File

@ -21,7 +21,7 @@ tf_class {
}
member_method {
name: "__init__"
argspec: "args=[\'self\', \'model_fn\', \'model_dir\', \'config\', \'params\', \'use_tpu\', \'train_batch_size\', \'eval_batch_size\', \'predict_batch_size\', \'batch_axis\', \'eval_on_tpu\', \'export_to_tpu\', \'export_to_cpu\', \'warm_start_from\', \'experimental_export_device_assignment\', \'embedding_config_spec\', \'export_saved_model_api_version\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'None\', \'True\', \'None\', \'None\', \'None\', \'None\', \'True\', \'True\', \'True\', \'None\', \'False\', \'None\', \'ExportSavedModelApiVersion.V1\'], "
argspec: "args=[\'self\', \'model_fn\', \'model_dir\', \'config\', \'params\', \'use_tpu\', \'train_batch_size\', \'eval_batch_size\', \'predict_batch_size\', \'batch_axis\', \'eval_on_tpu\', \'export_to_tpu\', \'export_to_cpu\', \'warm_start_from\', \'embedding_config_spec\', \'export_saved_model_api_version\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'None\', \'True\', \'None\', \'None\', \'None\', \'None\', \'True\', \'True\', \'True\', \'None\', \'None\', \'ExportSavedModelApiVersion.V1\'], "
}
member_method {
name: "eval_dir"