Home
last modified time | relevance | path

Searched refs:_distribution_strategy_stack (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Ddistribution_strategy_context.py62 …ops.get_default_graph()._distribution_strategy_stack.append(context) # pylint: disable=protected-…
66 ops.get_default_graph()._distribution_strategy_stack.pop(-1) # pylint: disable=protected-access
82 …return ops.get_default_graph()._distribution_strategy_stack[-1] # pylint: disable=protected-access
/external/tensorflow/tensorflow/python/framework/
Dfunc_graph.py379 old_strategy_stack = self._distribution_strategy_stack
380 self._distribution_strategy_stack = list(
381 graph._distribution_strategy_stack)
396 (self._distribution_strategy_stack and
421 self._distribution_strategy_stack = old_strategy_stack
Dops.py5181 def _distribution_strategy_stack(self): member in Graph
5184 self._thread_local._distribution_strategy_stack = [] # pylint: disable=protected-access
5185 return self._thread_local._distribution_strategy_stack # pylint: disable=protected-access
5187 @_distribution_strategy_stack.setter
5188 def _distribution_strategy_stack(self, _distribution_strategy_stack): argument
5189 self._thread_local._distribution_strategy_stack = ( # pylint: disable=protected-access
5190 _distribution_strategy_stack)
/external/tensorflow/tensorflow/python/eager/
Dfunction.py3201 strategy_stack = default_graph._distribution_strategy_stack