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

  /external/tensorflow/tensorflow/contrib/compiler/
xla.py 560 class _ModelFnWrapper(object):
561 """_ModelFnWrapper supports executing model_fn with XLA."""
788 return tf_decorator.make_decorator(function, _ModelFnWrapper(function))
  /external/tensorflow/tensorflow/python/tpu/
tpu_estimator.py     [all...]

Completed in 660 milliseconds