Home
last modified time | relevance | path

Searched refs:GetOneStringParameter (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/tools/graph_transforms/
Dset_device.cc27 TF_RETURN_IF_ERROR(context.GetOneStringParameter("device", "", &new_device)); in SetDevice()
Dtransform_utils.cc624 Status TransformFuncContext::GetOneStringParameter(const string& name, in GetOneStringParameter() function in tensorflow::graph_transforms::TransformFuncContext
650 TF_RETURN_IF_ERROR(GetOneStringParameter(name, "", &string_value)); in GetOneInt32Parameter()
667 TF_RETURN_IF_ERROR(GetOneStringParameter(name, "", &string_value)); in GetOneInt64Parameter()
684 TF_RETURN_IF_ERROR(GetOneStringParameter(name, "", &string_value)); in GetOneFloatParameter()
702 TF_RETURN_IF_ERROR(GetOneStringParameter(name, "", &string_value)); in GetOneBoolParameter()
Dinsert_logging.cc56 TF_RETURN_IF_ERROR(context.GetOneStringParameter("message", "", &message)); in InsertLogging()
Dfreeze_requantization_ranges.cc108 context.GetOneStringParameter("min_max_log_file", "", &min_max_log_file)); in FreezeRequantizationRanges()
Dtransform_utils.h231 Status GetOneStringParameter(const string& name, const string& default_value,
Dtransform_utils_test.cc980 TF_EXPECT_OK(context.GetOneStringParameter("bar", "d", &value)); in TestGetOneStringParameter()
982 EXPECT_FALSE(context.GetOneStringParameter("foo", "d", &value).ok()); in TestGetOneStringParameter()
983 TF_EXPECT_OK(context.GetOneStringParameter("not_present", "d", &value)); in TestGetOneStringParameter()