Searched defs:grad_state (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/python/compiler/xla/ | ||
D | xla.py | 296 def grad_state(self): member in XLACompileContext |
/external/tensorflow/tensorflow/python/ops/ | ||
D | control_flow_ops.py | 700 def grad_state(self): member in ControlFlowContext |
/external/tensorflow/tensorflow/python/tpu/ | ||
D | tpu.py | 674 def grad_state(self): member in TPUReplicateContext |