Home
last modified time | relevance | path

Searched refs:_get_control_flow_context (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/python/ops/
Dcontrol_flow_util.py64 ctxt = op._get_control_flow_context() # pylint: disable=protected-access
69 ctxt = graph._get_control_flow_context() # pylint: disable=protected-access
83 ctxt = op._get_control_flow_context() # pylint: disable=protected-access
88 ctxt = op._get_control_flow_context() # pylint: disable=protected-access
125 ctxt = c._get_control_flow_context() # pylint: disable=protected-access
153 ctxt = op._get_control_flow_context() # pylint: disable=protected-access
161 ctxt = op._get_control_flow_context() # pylint: disable=protected-access
182 ctxt = op._get_control_flow_context() # pylint: disable=protected-access
259 …return IsContainingContext(op._get_control_flow_context(), ctxt) # pylint: disable=protected-acce…
280 op_ctxt = op._get_control_flow_context() # pylint: disable=protected-access
[all …]
Dcontrol_flow_grad.py44 op_ctxt = op._get_control_flow_context()
45 grad_ctxt = graph._get_control_flow_context()
102 grad_ctxt = graph._get_control_flow_context()
151 op_ctxt = op._get_control_flow_context()
152 grad_ctxt = graph._get_control_flow_context()
211 grad_ctxt = graph._get_control_flow_context()
Dcontrol_flow_state.py60 …curr_ctxt = ops.get_default_graph()._get_control_flow_context() # pylint: disable=protected-access
77 max_iter_ctxt = max_iter.op._get_control_flow_context()
333 …curr_ctxt = ops.get_default_graph()._get_control_flow_context() # pylint: disable=protected-access
404 history_ctxt = history_value.op._get_control_flow_context()
407 value_ctxt = value.op._get_control_flow_context()
503 forward_ctxt = op._get_control_flow_context() # pylint: disable=protected-access
608 forward_ctxt = val.op._get_control_flow_context()
671 op_ctxt = op._get_control_flow_context()
787 op_ctxt = op._get_control_flow_context() # pylint: disable=protected-access
Dnumerics.py114 if op._get_control_flow_context() is not None: # pylint: disable=protected-access
Dcontrol_flow_util_v2.py215 control_flow_context = ops.get_default_graph()._get_control_flow_context()
Dcontrol_flow_ops.py642 self._outer_context = ops.get_default_graph()._get_control_flow_context()
733 self._context_stack.append(graph._get_control_flow_context())
1657 grad_ctxt = ops.get_default_graph()._get_control_flow_context()
1709 grad_ctxt = ops.get_default_graph()._get_control_flow_context()
1715 op_input_ctxt = op.inputs[0].op._get_control_flow_context()
Dlookup_ops.py175 …ops.get_default_graph()._get_control_flow_context() is not None): # pylint: disable=protected-acc…
429 …ops.get_default_graph()._get_control_flow_context() is not None): # pylint: disable=protected-acc…
Dspecial_math_ops.py313 context = ops.get_default_graph()._get_control_flow_context()
Dgradients_util.py637 ops.get_default_graph()._get_control_flow_context()):
Dcontrol_flow_ops_test.py477 c = op._get_control_flow_context()
491 control_flow_context = op._get_control_flow_context()
Drnn.py141 ctxt = ops.get_default_graph()._get_control_flow_context() # pylint: disable=protected-access
Dvariables.py1818 if self._initial_value.op._get_control_flow_context() is not None:
/external/tensorflow/tensorflow/python/tpu/
Dtpu.py181 context_ = graph._get_control_flow_context()
346 saved_context = graph._get_control_flow_context()
497 ctxt = x._get_control_flow_context()
758 initial_context = graph._get_control_flow_context() # pylint: disable=protected-access
769 final_context = graph._get_control_flow_context() # pylint: disable=protected-access
1748 context = graph._get_control_flow_context() # pylint: disable=protected-access
Dtensor_tracer.py1470 current_control_flow_context = graph._get_control_flow_context()
/external/tensorflow/tensorflow/python/framework/
Dsubscribe.py279 context = tensor.op._get_control_flow_context()
Dsubscribe_test.py364 return tensor.op._get_control_flow_context()
Dops.py1809 self._control_flow_context = self.graph._get_control_flow_context()
1888 def _get_control_flow_context(self): member in Operation
3000 def _get_control_flow_context(self): member in Graph
4411 self._old_control_flow_context = self._graph._get_control_flow_context()
Dops_test.py1117 self.assertIsNotNone(op._get_control_flow_context())
1118 self.assertEqual(op._get_control_flow_context().name,
1147 self.assertIsNotNone(op._get_control_flow_context())
1148 self.assertEqual(op._get_control_flow_context().name,
1197 self.assertIsNotNone(op.control_inputs[0]._get_control_flow_context())
/external/tensorflow/tensorflow/python/distribute/
Done_device_strategy.py329 ops.get_default_graph()._get_control_flow_context()) # pylint: disable=protected-access
Dtpu_strategy.py410 ops.get_default_graph()._get_control_flow_context()) # pylint: disable=protected-access
Dmirrored_strategy.py689 ops.get_default_graph()._get_control_flow_context()) # pylint: disable=protected-access
Dvalues.py938 context_ = graph._get_control_flow_context()
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dcontrol_flow_ops.py124 control_flow_context = g._get_control_flow_context() # pylint: disable=protected-access
/external/tensorflow/tensorflow/python/compiler/xla/
Dxla.py178 ctxt = x._get_control_flow_context()
/external/tensorflow/tensorflow/python/eager/
Dbackprop.py116 def _get_control_flow_context(self): member in _MockOp

12