HomeSort by relevance Sort by last modified time
    Searched refs:tpu_context (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/contrib/tpu/python/tpu/
tpu_context.py 22 from tensorflow.python.tpu.tpu_context import *
keras_tpu_variables.py 79 tpu_context = _enclosing_tpu_context()
80 if tpu_context is None:
83 return tpu_context.get_replicated_var_handle(self._name, self._vars)
  /external/tensorflow/tensorflow/python/distribute/
values.py 757 tpu_context = ops.get_default_graph()._get_control_flow_context()
759 while tpu_context is not None and not isinstance(
760 tpu_context, control_flow_ops.XLAControlFlowContext):
761 tpu_context = tpu_context.outer_context
762 return tpu_context
    [all...]
  /external/tensorflow/tensorflow/python/tpu/
tpu_estimator.py 69 from tensorflow.python.tpu import tpu_context
743 user_context = tpu_context.TPUContext(
782 user_context = tpu_context.TPUContext(
    [all...]

Completed in 83 milliseconds