Searched refs:_lookup_infeed_manager (Results 1 – 1 of 1) sorted by relevance
1137 def _lookup_infeed_manager(self, inputs): member in TPUFunction1271 infeed_manager = self._lookup_infeed_manager(inputs)1315 next_step_infeed_manager = self._lookup_infeed_manager(next_step_inputs)1316 cur_step_infeed_manager = self._lookup_infeed_manager(cur_step_inputs)