Searched refs:_capture_helper (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | cond_v2.py | 951 def _capture_helper(self, tensor, name): member in _CondGradFuncGraph
955 return super(_CondGradFuncGraph, self)._capture_helper(tensor, name)
974 return super(_CondGradFuncGraph, self)._capture_helper(tensor, name)
995 captured_tensor = super(_CondGradFuncGraph, self)._capture_helper(
1016 self)._capture_helper(optional, name)
|
| D | while_v2.py | 1131 def _capture_helper(self, tensor, name): member in _WhileBodyGradFuncGraph
1139 captured_tensor = super(_WhileBodyGradFuncGraph, self)._capture_helper(
1164 self)._capture_helper(tensor, name)
1230 captured_accumulator = super(_WhileBodyGradFuncGraph, self)._capture_helper(
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | func_graph.py | 773 return self._capture_helper(tensor, name, shape)
802 return self._capture_helper(tensor, name)
805 def _capture_helper(self, tensor, name, shape=None): member in FuncGraph
1007 return self._capture_helper(tensor, name)
|