Searched refs:colocation_stack (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/framework/ |
| D | function.py | 909 colocation_stack=None, argument
954 if colocation_stack is not None:
955 func_graph._colocation_stack = colocation_stack
|
| D | ops.py | 5153 def _colocation_stack(self, colocation_stack): argument
5156 self._thread_local._colocation_stack = colocation_stack
5159 self._graph_colocation_stack = colocation_stack
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | function.py | 3166 (parent_graph, device_functions, colocation_stack, in_cross_replica_context,
3170 colocation_stack, in_cross_replica_context, variable_policy,
3207 colocation_stack = ()
3213 colocation_stack = tuple(default_graph._colocation_stack.peek_objs())
3235 return (parent_graph, device_functions, colocation_stack,
|