Searched refs:TPUReplicateContext (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | tpu_util.py | 29 if isinstance(context_, tpu.TPUReplicateContext):
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_test.py | 48 context = tpu.TPUReplicateContext(b"context", 1, pivot=pivot)
64 context = tpu.TPUReplicateContext(b"context", 1, pivot=pivot)
|
| D | tpu.py | 205 if isinstance(context_, TPUReplicateContext):
266 class TPUReplicateContext(control_flow_ops.XLAControlFlowContext): class
294 super(TPUReplicateContext, self).__init__()
538 super(TPUReplicateContext, self).Enter()
831 if isinstance(context, TPUReplicateContext):
846 if isinstance(context, TPUReplicateContext):
1395 context = TPUReplicateContext(
|
| D | tpu_embedding_v2.py | 1046 if isinstance(ctx, tpu.TPUReplicateContext):
|