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

  /external/tensorflow/tensorflow/contrib/compiler/
xla.py 675 model_fn_args = function_utils.fn_args(self._model_fn)
678 if 'labels' in model_fn_args:
683 if 'mode' in model_fn_args:
686 if 'params' in model_fn_args:
  /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
estimator.py     [all...]
  /external/tensorflow/tensorflow/python/tpu/
tpu_estimator.py     [all...]

Completed in 1939 milliseconds