Home
last modified time | relevance | path

Searched defs:model_fn (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/contrib/distribute/python/
Dmirrored_strategy_multigpu_test.py229 def model_fn(): function
265 def model_fn(): function
286 def model_fn(): function
298 def model_fn(): function
325 def model_fn(): function
338 def model_fn(): function
351 def model_fn(): function
361 def model_fn(): function
374 def model_fn(): function
394 def model_fn(): function
[all …]
Dsingle_loss_example.py72 def model_fn(x): function
109 def model_fn(x): function
Dkeras_optimizer_v2_test.py63 def model_fn(): function
Dparameter_server_strategy_test.py164 def model_fn(): function
271 def model_fn(): function
310 def model_fn(): function
420 def model_fn(): function
Dminimize_loss_test.py294 def model_fn(inputs): function
386 def model_fn(output_context, x): function
Dcollective_all_reduce_strategy_test.py227 def model_fn(): function
287 def model_fn(): function
Dkeras_correctness_test_base.py233 def fit_eval_and_predict(initial_weights, input_fn, model_fn, argument
Dstrategy_test_lib.py342 def model_fn(): function
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Destimators_test.py59 def model_fn(features, labels): function
99 def model_fn(features, labels): function
140 def model_fn(features, labels): function
Dlogistic_regressor.py38 def _get_model_fn_with_logistic_metrics(model_fn): argument
79 model_fn, thresholds=None, model_dir=None, config=None, argument
Destimator.py474 def model_fn(self): member in BaseEstimator
1111 model_fn=None, argument
/external/tensorflow/tensorflow/python/keras/applications/
Dapplications_test.py46 def test_feature_extration_model(self, model_fn, output_dim): argument
/external/tensorflow/tensorflow/contrib/distribute/python/examples/
Dsimple_estimator_example.py34 def model_fn(features, labels, mode): # pylint: disable=unused-argument function
/external/tensorflow/tensorflow/contrib/eager/python/examples/revnet/
Dmain_estimator.py30 def model_fn(features, labels, mode, params): function
Dmain_estimator_tpu.py101 def model_fn(features, labels, mode, params): function
/external/tensorflow/tensorflow/contrib/gan/python/estimator/python/
Dlatent_gan_estimator_impl.py96 def model_fn(features, labels, mode, params): function
/external/tensorflow/tensorflow/contrib/compiler/
Dxla_test.py308 def test_compile(self, model_fn, params): argument
332 def test_not_compile(self, model_fn, params): argument
558 def model_fn(features, labels, mode, params): function
580 def model_fn(features, labels, mode, params): function
/external/tensorflow/tensorflow/python/keras/layers/
Dtensorflow_op_layer_test.py156 def test_autolambda(self, model_fn): argument
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dgradients_test.py170 def model_fn(activation): function
194 def model_fn(inps, init_state): function
/external/tensorflow/tensorflow/contrib/factorization/python/ops/
Dkmeans.py154 def model_fn(self, features, mode, config): member in _ModelFn
/external/tensorflow/tensorflow/python/tpu/
Dtpu_estimator.py1405 def __init__(self, model_fn, config, params, ctx): argument
2220 model_fn=None, argument
2766 def _augment_model_fn(self, model_fn, batch_axis): argument