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.py109 def model_fn(): function
127 def model_fn(): function
143 def model_fn(): function
157 def model_fn(): function
168 def model_fn(): function
182 def model_fn(): function
203 def model_fn(): function
217 def model_fn(): function
256 def model_fn(): function
433 def model_fn(name): function
[all …]
Dmirrored_strategy_test.py365 def model_fn(): function
403 def model_fn(): function
425 def model_fn(): function
439 def model_fn(): function
461 def model_fn(): function
478 def model_fn(): function
497 def model_fn(): function
510 def model_fn(): function
538 def model_fn(): function
556 def model_fn(): function
[all …]
Dstrategy_common_test.py83 def model_fn(): function
94 def model_fn(): function
105 def model_fn(): function
117 def model_fn(): function
146 def model_fn(): function
158 def model_fn(): function
171 def model_fn(): function
187 def model_fn(): function
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.py194 def model_fn(): function
Dstrategy_test_lib.py421 def model_fn(): function
/external/tensorflow/tensorflow/python/keras/distribute/
Dtest_example.py38 def model_fn(x): function
75 def model_fn(x): function
Dkeras_optimizer_v2_test.py59 def model_fn(): function
Dmirrored_variable_test.py88 def model_fn(features): function
Dminimize_loss_test.py337 def model_fn(inputs): function
435 def model_fn(output_context, x): function
Dkeras_correctness_test_base.py252 model_fn, argument
Ddistribute_strategy_test.py2550 def test_fit_and_evaluate(self, distribution, model_fn, l1, l2): argument
/external/tensorflow/tensorflow/python/keras/benchmarks/
Doptimizer_benchmarks_test.py27 def model_fn(): function
Dbenchmark_util.py95 def measure_performance(model_fn, argument
/external/tensorflow/tensorflow/python/keras/tests/
Dmodel_architectures_test.py73 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/python/keras/applications/
Dapplications_test.py140 def _get_output_shape(model_fn): argument
/external/tensorflow/tensorflow/core/platform/
Dram_file_system_test.py116 def model_fn(features, labels, mode, params): function
/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.py316 def test_autolambda(self, model_fn): argument
/external/tensorflow/tensorflow/python/keras/saving/
Dsave_test.py1202 def test_functional(self, model_fn): argument