Esempio n. 1
0
 def parameter_modifiers(self):
     return [
         combinations_lib.OptionalParameter("required_tpus"),
         combinations_lib.OptionalParameter("required_tpu"),
         combinations_lib.OptionalParameter("use_cloud_tpu"),
     ]
Esempio n. 2
0
 def parameter_modifiers(self):
   return [test_combinations.OptionalParameter('tf_function')]
Esempio n. 3
0
 def parameter_modifiers(self):
     return [
         DistributionParameter(),
         combinations_lib.OptionalParameter("use_var_policy"),
     ]
Esempio n. 4
0
 def parameter_modifiers(self):
     return [combinations_lib.OptionalParameter("required_gpus")]
Esempio n. 5
0
 def parameter_modifiers(self):
     return [test_combinations.OptionalParameter('model_type')]
Esempio n. 6
0
 def parameter_modifiers(self):
     return [test_combinations.OptionalParameter('run_eagerly')]
Esempio n. 7
0
 def parameter_modifiers(self):
   return [test_combinations.OptionalParameter("tf_api_version")]
Esempio n. 8
0
 def parameter_modifiers(self):
   return [test_combinations.OptionalParameter("mode")]
Esempio n. 9
0
 def parameter_modifiers(self):
     return [
         test_combinations.OptionalParameter("required_tpus"),
         test_combinations.OptionalParameter("required_tpu"),
         test_combinations.OptionalParameter("use_cloud_tpu"),
     ]