Home
last modified time | relevance | path

Searched refs:TPUReplicateContext (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dtpu_util.py29 if isinstance(context_, tpu.TPUReplicateContext):
/external/tensorflow/tensorflow/python/tpu/
Dtpu_test.py48 context = tpu.TPUReplicateContext(b"context", 1, pivot=pivot)
64 context = tpu.TPUReplicateContext(b"context", 1, pivot=pivot)
Dtpu.py205 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(
Dtpu_embedding_v2.py1046 if isinstance(ctx, tpu.TPUReplicateContext):