コード例 #1
0
 def parameter_modifiers(self):
     return [
         combinations_lib.OptionalParameter("required_tpus"),
         combinations_lib.OptionalParameter("required_tpu"),
         combinations_lib.OptionalParameter("use_cloud_tpu"),
     ]
コード例 #2
0
ファイル: test_util.py プロジェクト: qoffee/probability 
 def parameter_modifiers(self):
   return [test_combinations.OptionalParameter('tf_function')]
コード例 #3
0
 def parameter_modifiers(self):
     return [
         DistributionParameter(),
         combinations_lib.OptionalParameter("use_var_policy"),
     ]
コード例 #4
0
 def parameter_modifiers(self):
     return [combinations_lib.OptionalParameter("required_gpus")]
コード例 #5
0
 def parameter_modifiers(self):
     return [test_combinations.OptionalParameter('model_type')]
コード例 #6
0
 def parameter_modifiers(self):
     return [test_combinations.OptionalParameter('run_eagerly')]
コード例 #7
0
 def parameter_modifiers(self):
   return [test_combinations.OptionalParameter("tf_api_version")]
コード例 #8
0
 def parameter_modifiers(self):
   return [test_combinations.OptionalParameter("mode")]
コード例 #9
0
 def parameter_modifiers(self):
     return [
         test_combinations.OptionalParameter("required_tpus"),
         test_combinations.OptionalParameter("required_tpu"),
         test_combinations.OptionalParameter("use_cloud_tpu"),
     ]