Searched refs:rewrite_for_inference (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ | ||
| D | tpu.py | 1514 def rewrite_for_inference(computation, function |
| D | tpu_estimator.py | 2498 tensors_on_cpu = tpu.rewrite_for_inference( |