Commit d0cc3477 authored by Russell Power's avatar Russell Power Committed by TensorFlower Gardener
Browse files

TPUEstimator: Add export_to_cpu option to enable/disable CPU export. 

Not all TPU ops are supported on the CPU (in particular some bfloat16
operations).  These models will fail to export on the CPU.  This CL adds a flag
to allow users to disable CPU export in this case.

PiperOrigin-RevId: 229780875
parent c6073044
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment