Searched defs:tpu_assignment (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/contrib/tpu/python/tpu/ |
| D | keras_support.py | 645 def __init__(self, tpu_assignment): argument
726 def __init__(self, dataset, tpu_assignment, mode): argument
851 def _inject_tpu_inputs_for_dataset(tpu_assignment, mode, argument
878 def _inject_tpu_inputs_for_infeed(tpu_assignment, mode, argument
927 def __init__(self, model, execution_mode, tpu_assignment): argument
|