Home
last modified time | relevance | path

Searched refs:test_combinations (Results 1 – 25 of 57) sorted by relevance

123

/external/tensorflow/tensorflow/python/framework/
Dcombinations.py30 from tensorflow.python.framework import test_combinations
34 class EagerGraphCombination(test_combinations.TestCombination):
54 return [test_combinations.OptionalParameter("mode")]
57 class TFVersionCombination(test_combinations.TestCombination):
76 return [test_combinations.OptionalParameter("tf_api_version")]
80 test_combinations.generate,
81 test_combinations=(EagerGraphCombination(), TFVersionCombination()))
82 combine = test_combinations.combine
83 times = test_combinations.times
84 NamedObject = test_combinations.NamedObject
Dtest_combinations.py234 def generate(combinations, test_combinations=()): argument
284 test_method, test_combinations=test_combinations),
292 test_method_or_class, test_combinations=test_combinations)
298 def _augment_with_special_arguments(test_method, test_combinations): argument
306 for combination in test_combinations:
316 for combination in test_combinations:
353 for combination in test_combinations:
/external/tensorflow/tensorflow/python/keras/
Dcombinations.py25 from tensorflow.python.framework import test_combinations
64 class KerasModeCombination(test_combinations.TestCombination):
79 return [test_combinations.OptionalParameter('run_eagerly')]
82 class KerasModelTypeCombination(test_combinations.TestCombination):
100 return [test_combinations.OptionalParameter('model_type')]
106 test_combinations=_defaults +
108 combine = test_combinations.combine
109 times = test_combinations.times
110 NamedObject = test_combinations.NamedObject
/external/tensorflow/tensorflow/python/training/experimental/
Dloss_scaling_gradient_tape_test.py29 from tensorflow.python.framework import test_combinations
82 @test_combinations.generate(test_combinations.combine(
102 @test_combinations.generate(test_combinations.combine(
122 @test_combinations.generate(test_combinations.combine(
154 @test_combinations.generate(test_combinations.combine(
175 @test_combinations.generate(test_combinations.combine(
220 @test_combinations.generate(test_combinations.combine(
246 @test_combinations.generate(test_combinations.combine(
258 @test_combinations.generate(test_combinations.combine(
271 @test_combinations.generate(test_combinations.combine(
[all …]
/external/tensorflow/tensorflow/python/distribute/
Dcombinations_test.py47 test_combinations=(combinations.ClusterCombination(),))
56 test_combinations=(combinations.ClusterCombination(),))
63 test_combinations=(combinations.ClusterCombination(),))
74 test_combinations=(combinations.ClusterCombination(),))
88 test_combinations=(combinations.ClusterCombination(),))
131 test_combinations=(combinations.ClusterCombination(),))
Dcombinations.py43 from tensorflow.python.framework import test_combinations as combinations_lib
329 def generate(combinations, test_combinations=()): argument
352 combinations, test_combinations=default_combinations + test_combinations)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.__internal__.test.combinations.-optional-parameter.pbtxt3 is_instance: "<class \'tensorflow.python.framework.test_combinations.OptionalParameter\'>"
4 is_instance: "<class \'tensorflow.python.framework.test_combinations.ParameterModifier\'>"
Dtensorflow.__internal__.test.combinations.-named-object.pbtxt3 is_instance: "<class \'tensorflow.python.framework.test_combinations.NamedObject\'>"
Dtensorflow.__internal__.test.combinations.-parameter-modifier.pbtxt3 is_instance: "<class \'tensorflow.python.framework.test_combinations.ParameterModifier\'>"
Dtensorflow.__internal__.test.combinations.-test-combination.pbtxt3 is_instance: "<class \'tensorflow.python.framework.test_combinations.TestCombination\'>"
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
Dnoop_elimination_test.py85 test_combinations = functools.reduce(reduce_fn, cases, [])
87 return test_combinations
/external/tensorflow/tensorflow/python/keras/distribute/
Dmodel_combinations.py21 from tensorflow.python.framework import test_combinations as combinations
Dkeras_models_test.py26 from tensorflow.python.framework import test_combinations as combinations
Dworker_training_state_test.py26 from tensorflow.python.framework import test_combinations as combinations
Dsaved_model_mixed_api_test.py28 from tensorflow.python.framework import test_combinations as combinations
Dkeras_save_load_test.py22 from tensorflow.python.framework import test_combinations as combinations
Dkeras_stateful_lstm_model_correctness_test.py24 from tensorflow.python.framework import test_combinations as combinations
Dcollective_all_reduce_strategy_test.py30 from tensorflow.python.framework import test_combinations as combinations
Dmirrored_variable_test.py29 from tensorflow.python.framework import test_combinations as combinations
/external/tensorflow/tensorflow/python/keras/layers/preprocessing/
Ddiscretization_distribution_test.py26 from tensorflow.python.framework import test_combinations as combinations
Dhashing_distribution_test.py28 from tensorflow.python.framework import test_combinations as combinations
Dindex_lookup_distribution_test.py29 from tensorflow.python.framework import test_combinations as combinations
Dimage_preprocessing_distribution_test.py27 from tensorflow.python.framework import test_combinations as combinations
Dcategory_encoding_distribution_test.py29 from tensorflow.python.framework import test_combinations as combinations
Dcategory_crossing_distribution_test.py29 from tensorflow.python.framework import test_combinations as combinations

123