8000 Add option to TPUEstimator to only perform outfeed every n steps duri… · compnerd/tensorflow@16b78f4 · GitHub
[go: up one dir, main page]

Skip to content

Commit 16b78f4

Browse files
Bruce Fontainetensorflower-gardener
authored andcommitted
Add option to TPUEstimator to only perform outfeed every n steps during a TPU training loop.
PiperOrigin-RevId: 252859456
1 parent bb7f323 commit 16b78f4

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tensorflow/tools/api/golden/v1/tensorflow.estimator.tpu.-t-p-u-config.pbtxt

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,10 @@ tf_class {
77
name: "eval_training_input_configuration"
88
mtype: "<type \'property\'>"
99
}
10+
member {
11+
name: "experimental_host_call_every_n_steps"
12+
mtype: "<type \'property\'>"
13+
}
1014
member {
1115
name: "initial_infeed_sleep_secs"
1216
mtype: "<type \'property\'>"

0 commit comments

Comments
 (0)
0