Searched refs:_tpu_model_ops_for_input_specs (Results 1 – 1 of 1) sorted by relevance
1159 def _tpu_model_ops_for_input_specs(self, input_specs, infeed_manager): member in TPUFunction1276 tpu_model_ops = self._tpu_model_ops_for_input_specs(input_specs,1345 cur_tpu_model_ops = self._tpu_model_ops_for_input_specs(1352 next_tpu_model_ops = self._tpu_model_ops_for_input_specs(