Searched refs:tpu_run (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | tpu_strategy.py | 402 return self.extended.tpu_run(fn, args, kwargs, options)
665 return self.extended.tpu_run(fn, args, kwargs, options)
779 return self.extended.tpu_run(fn, args, kwargs, options)
1428 def tpu_run(self, fn, args, kwargs, options=None): member in TPUExtended
|