Home
last modified time | relevance | path

Searched refs:ds_combinations (Results 1 – 25 of 36) sorted by relevance

12

/external/tensorflow/tensorflow/python/keras/distribute/
Dsaved_model_save_load_test.py22 from tensorflow.python.distribute import combinations as ds_combinations unknown
57 @ds_combinations.generate(test_base.simple_models_with_strategies())
63 @ds_combinations.generate(
71 @ds_combinations.generate(
83 @ds_combinations.generate(
133 @ds_combinations.generate(test_base.tfmodule_models_with_strategies())
139 @ds_combinations.generate(
147 @ds_combinations.generate(
159 @ds_combinations.generate(
Dkeras_utils_test.py29 from tensorflow.python.distribute import combinations as ds_combinations unknown
79 @ds_combinations.generate(
133 @ds_combinations.generate(
157 @ds_combinations.generate(
187 @ds_combinations.generate(
211 @ds_combinations.generate(
235 @ds_combinations.generate(
285 @ds_combinations.generate(
319 @ds_combinations.generate(
346 @ds_combinations.generate(
[all …]
Dkeras_dnn_correctness_test.py23 from tensorflow.python.distribute import combinations as ds_combinations unknown
107 @ds_combinations.generate(
113 @ds_combinations.generate(
123 @ds_combinations.generate(
138 @ds_combinations.generate(all_strategy_combinations_with_graph_mode())
177 @ds_combinations.generate(
226 @ds_combinations.generate(
274 @ds_combinations.generate(
294 @ds_combinations.generate(all_strategy_combinations_with_graph_mode())
313 @ds_combinations.generate(
Dsaved_model_mixed_api_test.py26 from tensorflow.python.distribute import combinations as ds_combinations unknown
57 @ds_combinations.generate(test_base.simple_models_with_strategies())
63 @ds_combinations.generate(
71 @ds_combinations.generate(
Dkeras_image_model_correctness_test.py22 from tensorflow.python.distribute import combinations as ds_combinations unknown
102 @ds_combinations.generate(
108 @ds_combinations.generate(
119 @ds_combinations.generate(
133 @ds_combinations.generate(
148 @ds_combinations.generate(
Dkeras_save_load_test.py20 from tensorflow.python.distribute import combinations as ds_combinations unknown
49 @ds_combinations.generate(test_base.simple_models_with_strategies())
55 @ds_combinations.generate(
63 @ds_combinations.generate(
Dcustom_training_loop_metrics_test.py25 from tensorflow.python.distribute import combinations as ds_combinations unknown
37 @ds_combinations.generate(
62 @ds_combinations.generate(
86 @ds_combinations.generate(
103 @ds_combinations.generate(
Ddistribute_strategy_test.py33 from tensorflow.python.distribute import combinations as ds_combinations unknown
370 @ds_combinations.generate(all_strategy_combinations())
391 @ds_combinations.generate(all_strategy_combinations())
434 @ds_combinations.generate(all_strategy_combinations())
456 @ds_combinations.generate(all_strategy_combinations())
471 @ds_combinations.generate(all_strategy_combinations())
505 @ds_combinations.generate(all_strategy_combinations())
553 @ds_combinations.generate(all_strategy_combinations())
596 @ds_combinations.generate(
614 @ds_combinations.generate(all_strategy_combinations())
[all …]
Dmulti_worker_callback_tf2_test.py28 from tensorflow.python.distribute import combinations as ds_combinations unknown
89 @ds_combinations.generate(
147 @ds_combinations.generate(combinations.combine(mode=['eager']))
173 @ds_combinations.generate(combinations.combine(mode=['eager']))
240 @ds_combinations.generate(combinations.combine(mode=['eager']))
278 @ds_combinations.generate(combinations.combine(mode=['eager']))
307 @ds_combinations.generate(combinations.combine(mode=['eager']))
336 @ds_combinations.generate(combinations.combine(mode=['eager']))
Dcustom_training_loop_optimizer_test.py22 from tensorflow.python.distribute import combinations as ds_combinations unknown
36 @ds_combinations.generate(
74 @ds_combinations.generate(
101 @ds_combinations.generate(
Dkeras_rnn_model_correctness_test.py24 from tensorflow.python.distribute import combinations as ds_combinations unknown
86 @ds_combinations.generate(
107 @ds_combinations.generate(
114 @ds_combinations.generate(
Dkeras_stateful_lstm_model_correctness_test.py22 from tensorflow.python.distribute import combinations as ds_combinations unknown
85 @ds_combinations.generate(test_combinations_for_stateful_embedding_model())
94 @ds_combinations.generate(
Dkeras_embedding_model_correctness_test.py23 from tensorflow.python.distribute import combinations as ds_combinations unknown
59 @ds_combinations.generate(
68 @ds_combinations.generate(
151 @ds_combinations.generate(
Dmirrored_variable_test.py23 from tensorflow.python.distribute import combinations as ds_combinations unknown
43 @ds_combinations.generate(
47 ds_combinations.NamedDistribution(
Dkeras_premade_models_test.py23 from tensorflow.python.distribute import combinations as ds_combinations unknown
69 @ds_combinations.generate(strategy_combinations_eager_data_fn())
82 @ds_combinations.generate(strategy_combinations_eager_data_fn())
Dkeras_models_test.py25 from tensorflow.python.distribute import combinations as ds_combinations unknown
33 @ds_combinations.generate(
Dkeras_optimizer_v2_test.py24 from tensorflow.python.distribute import combinations as ds_combinations unknown
49 @ds_combinations.generate(
105 @ds_combinations.generate(
Dcheckpointing_test.py24 from tensorflow.python.distribute import combinations as ds_combinations unknown
40 @ds_combinations.generate(
99 @ds_combinations.generate(
Dworker_training_state_test.py24 from tensorflow.python.distribute import combinations as ds_combinations unknown
36 @ds_combinations.generate(
Dminimize_loss_test.py25 from tensorflow.python.distribute import combinations as ds_combinations unknown
74 @ds_combinations.generate(
125 @ds_combinations.generate(
164 @ds_combinations.generate(
231 @ds_combinations.generate(
298 @ds_combinations.generate(
414 @ds_combinations.generate(
539 @ds_combinations.generate(
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dautocast_variable_test.py27 from tensorflow.python.distribute import combinations as ds_combinations unknown
65 @ds_combinations.generate(combinations.combine(mode=['graph', 'eager']))
72 @ds_combinations.generate(maybe_distribute)
113 @ds_combinations.generate(maybe_distribute)
128 @ds_combinations.generate(maybe_distribute)
145 @ds_combinations.generate(maybe_distribute)
223 @ds_combinations.generate(maybe_distribute)
269 @ds_combinations.generate(maybe_distribute)
345 @ds_combinations.generate(maybe_distribute)
361 @ds_combinations.generate(maybe_distribute)
[all …]
/external/tensorflow/tensorflow/python/distribute/
Drandom_generator_test.py27 from tensorflow.python.distribute import combinations as ds_combinations unknown
78 @ds_combinations.generate(
101 @ds_combinations.generate(
132 @ds_combinations.generate(
161 @ds_combinations.generate(
208 @ds_combinations.generate(
/external/tensorflow/tensorflow/python/keras/layers/preprocessing/
Ddiscretization_distribution_test.py24 from tensorflow.python.distribute import combinations as ds_combinations unknown
34 @ds_combinations.generate(
Dhashing_distribution_test.py25 from tensorflow.python.distribute import combinations as ds_combinations unknown
36 @ds_combinations.generate(
Dindex_lookup_distribution_test.py25 from tensorflow.python.distribute import combinations as ds_combinations unknown
45 @ds_combinations.generate(

12