Searched refs:_inject_tpu_inputs_for_dataset (Results 1 – 1 of 1) sorted by relevance
782 _inject_tpu_inputs_for_dataset(851 def _inject_tpu_inputs_for_dataset(tpu_assignment, mode, function