Searched defs:EnterGradientColocation (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ | ||
| D | tpu.py | 366 def EnterGradientColocation(self, op, gradient_uid): member in TPUReplicateContext |
| /external/tensorflow/tensorflow/python/ops/ | ||
| D | control_flow_ops.py | 742 def EnterGradientColocation(self, op, gradient_uid): member in ControlFlowContext |