Home
last modified time | relevance | path

Searched refs:use_dataset (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/python/keras/layers/preprocessing/
Dnormalization_tpu_test.py88 for use_dataset in (True, False):
91 if use_dataset:
93 case["use_dataset"] = use_dataset
105 def test_layer_computation(self, adapt_data, axis, test_data, use_dataset, argument
108 if use_dataset:
Dnormalization_distribution_test.py97 for use_dataset in (True, False):
100 if use_dataset:
102 case["use_dataset"] = use_dataset
117 use_dataset, expected): argument
119 if use_dataset:
Dnormalization_test.py116 for use_dataset in (True, False):
119 if use_dataset:
121 case["use_dataset"] = use_dataset
145 def test_layer_computation(self, adapt_data, axis, test_data, use_dataset, argument
148 if use_dataset:
Dstring_lookup_test.py75 for use_dataset in (True, False):
78 if use_dataset:
80 case["use_dataset"] = use_dataset
92 use_dataset, expected_output, argument
98 if use_dataset:
Dinteger_lookup_test.py81 for use_dataset in (True, False):
84 if use_dataset:
86 case["use_dataset"] = use_dataset
98 use_dataset, expected_output, argument
104 if use_dataset:
Ddiscretization_test.py193 def test_layer_computation(self, adapt_data, test_data, use_dataset, argument
198 if use_dataset:
Dtext_vectorization_test.py273 for use_dataset in (True, False):
276 if use_dataset:
278 case["use_dataset"] = use_dataset
291 use_dataset, expected_output): argument
299 if use_dataset:
Dindex_lookup_test.py309 for use_dataset in (True, False):
312 if use_dataset:
314 case["use_dataset"] = use_dataset
326 use_dataset, expected_output, argument
338 if use_dataset:
/external/tensorflow/tensorflow/python/keras/utils/
Dcomposite_tensor_support_test.py299 def get_kwargs(use_dataset, action="predict"): argument
300 if use_dataset or not context.executing_eagerly():
308 def prepare_inputs(data, use_dict, use_dataset, action, input_name): argument
314 if use_dataset:
330 use_dataset=[True, False],
334 def test_sparse_tensors(self, use_dict, use_dataset, action): argument
354 kwargs = get_kwargs(use_dataset, action)
359 use_dataset, action,
495 use_dataset=[True, False],
500 def test_ragged_input(self, use_dict, use_dataset, action): argument
[all …]
/external/tensorflow/tensorflow/python/distribute/
Dcollective_all_reduce_strategy_test.py346 mode=['graph'], required_gpus=[0, 1, 2], use_dataset=[True, False]))
347 def testMakeInputFnIterator(self, required_gpus, use_dataset): argument
349 if use_dataset:
372 test_reinitialize=use_dataset,
373 ignore_order=not use_dataset)
446 mode=['graph'], required_gpus=2, use_dataset=[True, False]))
447 def testMakeInputFnIterator(self, required_gpus, use_dataset): argument
448 if use_dataset:
471 test_reinitialize=use_dataset,
472 ignore_order=not use_dataset)
Dparameter_server_strategy_test.py640 mode=['graph'], required_gpus=[1, 2], use_dataset=[True, False]))
641 def testMakeInputFnIteratorDistributed(self, required_gpus, use_dataset): argument
642 if use_dataset:
665 test_reinitialize=use_dataset,
666 ignore_order=not use_dataset)
670 mode=['graph'], required_gpus=[1, 2], use_dataset=[True, False]))
671 def testMakeInputFnIteratorLocal(self, required_gpus, use_dataset): argument
672 if use_dataset:
696 test_reinitialize=use_dataset,
697 ignore_order=not use_dataset)
/external/tensorflow/tensorflow/python/keras/saving/saved_model/
Dsaved_model_test.py127 def _test_save_and_load(self, use_dataset=False): argument
147 if use_dataset:
178 self._test_save_and_load(use_dataset=True)
182 self._test_save_and_load(use_dataset=True)