Searched refs:_specialize_model (Results 1 – 1 of 1) sorted by relevance
939 def _specialize_model(self, input_specs, infeed_manager): member in TPUFunction1185 new_tpu_model_ops = self._specialize_model(input_specs,