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