def parameter_modifiers(self): return [ combinations_lib.OptionalParameter("required_tpus"), combinations_lib.OptionalParameter("required_tpu"), combinations_lib.OptionalParameter("use_cloud_tpu"), ]
def parameter_modifiers(self): return [test_combinations.OptionalParameter('tf_function')]
def parameter_modifiers(self): return [ DistributionParameter(), combinations_lib.OptionalParameter("use_var_policy"), ]
def parameter_modifiers(self): return [combinations_lib.OptionalParameter("required_gpus")]
def parameter_modifiers(self): return [test_combinations.OptionalParameter('model_type')]
def parameter_modifiers(self): return [test_combinations.OptionalParameter('run_eagerly')]
def parameter_modifiers(self): return [test_combinations.OptionalParameter("tf_api_version")]
def parameter_modifiers(self): return [test_combinations.OptionalParameter("mode")]
def parameter_modifiers(self): return [ test_combinations.OptionalParameter("required_tpus"), test_combinations.OptionalParameter("required_tpu"), test_combinations.OptionalParameter("use_cloud_tpu"), ]