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):
|