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