Skip to content
This repository was archived by the owner on Jul 10, 2025. It is now read-only.

Commit 0bfae5b

Browse files
authored
Update 20191106-tf2-tpu-savedmodel.md
1 parent 9008400 commit 0bfae5b

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

rfcs/20191106-tf2-tpu-savedmodel.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -231,6 +231,8 @@ def predict_step(image_tensors):
231231
- `num_batch_threads`, `max_batch_size`, `batch_timeout_micros`,
232232
`allowed_batch_sizes`, `max_enqueued_batches`: arguments used to
233233
configure BatchFunction.
234+
- `preserve_cpu_fn`: boolean. With this set to true, users avoid having to
235+
copy-paste the same block of code for CPU inference.
234236

235237
### Changes to Keras API
236238

0 commit comments

Comments
 (0)