Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/distribute/
Dmirrored_variable_test.py104 def model_fn(): function
122 def model_fn(): function
138 def model_fn(): function
152 def model_fn(): function
163 def model_fn(): function
177 def model_fn(): function
198 def model_fn(): function
212 def model_fn(): function
251 def model_fn(): function
428 def model_fn(name): function
[all …]
Dmirrored_strategy_test.py338 def model_fn(): function
376 def model_fn(): function
398 def model_fn(): function
412 def model_fn(): function
434 def model_fn(): function
451 def model_fn(): function
470 def model_fn(): function
483 def model_fn(): function
511 def model_fn(): function
529 def model_fn(): function
[all …]
Dsingle_loss_example.py65 def model_fn(x): function
102 def model_fn(x): function
Dparameter_server_strategy_test.py135 def model_fn(): function
241 def model_fn(): function
280 def model_fn(): function
386 def model_fn(): function
Dcollective_all_reduce_strategy_test.py193 def model_fn(): function
Dstrategy_test_lib.py421 def model_fn(): function
/external/tensorflow/tensorflow/python/keras/distribute/
Dtest_example.py42 def model_fn(x): function
79 def model_fn(x): function
Dmirrored_variable_test.py75 def model_fn(features): function
Dkeras_optimizer_v2_test.py62 def model_fn(): function
Dminimize_loss_test.py339 def model_fn(inputs): function
437 def model_fn(output_context, x): function
Dkeras_correctness_test_base.py256 model_fn, argument
Ddistribute_strategy_test.py2553 def test_fit_and_evaluate(self, distribution, model_fn, l1, l2): argument
/external/tensorflow/tensorflow/python/keras/integration_test/
Dvectorized_map_test.py29 def model_fn(arg): function
/external/tensorflow/tensorflow/python/keras/benchmarks/
Doptimizer_benchmarks_test.py30 def model_fn(): function
Dbenchmark_util.py98 def measure_performance(model_fn, argument
/external/tensorflow/tensorflow/python/keras/tests/
Dmodel_architectures_test.py76 def test_basic_saving_and_loading(self, model_fn): argument
/external/tensorflow/tensorflow/python/tpu/
Dasync_checkpoint_test.py65 def model_fn(features, labels, mode, params): function
/external/tensorflow/tensorflow/core/platform/
Dram_file_system_test.py101 def model_fn(features, labels, mode, params): function
/external/tensorflow/tensorflow/python/keras/applications/
Dapplications_test.py138 def _get_output_shape(model_fn): argument
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dgradients_test.py169 def model_fn(activation): function
193 def model_fn(inps, init_state): function
/external/tensorflow/tensorflow/python/keras/layers/
Dtensorflow_op_layer_test.py319 def test_autolambda(self, model_fn): argument
/external/tensorflow/tensorflow/python/keras/saving/
Dsave_test.py1062 def test_functional(self, model_fn): argument