Searched refs:_capture_helper (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | cond_v2.py | 910 def _capture_helper(self, tensor, name): member in _CondGradFuncGraph
914 return super(_CondGradFuncGraph, self)._capture_helper(tensor, name)
933 return super(_CondGradFuncGraph, self)._capture_helper(tensor, name)
954 captured_tensor = super(_CondGradFuncGraph, self)._capture_helper(
975 self)._capture_helper(optional, name)
|
| D | while_v2.py | 1082 def _capture_helper(self, tensor, name): member in _WhileBodyGradFuncGraph
1090 captured_tensor = super(_WhileBodyGradFuncGraph, self)._capture_helper(
1115 self)._capture_helper(tensor, name)
1181 captured_accumulator = super(_WhileBodyGradFuncGraph, self)._capture_helper(
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | func_graph.py | 634 return self._capture_helper(tensor, name, shape)
648 return self._capture_helper(tensor, name)
651 def _capture_helper(self, tensor, name, shape=None): member in FuncGraph
732 return self._capture_helper(tensor, name)
|