Searched refs:_update_config_proto (Results 1 – 5 of 5) sorted by relevance
224 config_proto = self._update_config_proto(config_proto)384 session_config.CopyFrom(self._update_config_proto(session_config))386 def _update_config_proto(self, config_proto): member in CollectiveAllReduceExtended
491 session_config.CopyFrom(self._update_config_proto(session_config))493 def _update_config_proto(self, config_proto): member in ParameterServerStrategyExtended
601 session_config.CopyFrom(self._update_config_proto(session_config))603 def _update_config_proto(self, config_proto): member in TPUExtended
599 return self._extended._update_config_proto(config_proto) # pylint: disable=protected-access1339 def _update_config_proto(self, config_proto): member in DistributionStrategyExtended
675 session_config.CopyFrom(self._update_config_proto(session_config))686 def _update_config_proto(self, config_proto): member in MirroredExtended