Skip to content

Commit

Permalink
Automated Code Change
Browse files Browse the repository at this point in the history 
PiperOrigin-RevId: 625985519
  • Loading branch information
@tensorflower-gardener
tensorflower-gardener committed Apr 18, 2024
1 parent 297b257 commit d709dd2
Show file tree
Hide file tree
Showing 2 changed files with 26 additions and 26 deletions.
40 changes: 20 additions & 20 deletions tensorflow/core/runtime_fallback/util/attr_util.cc
View file
Expand Up @@ -270,22 +270,22 @@ llvm::Error FillAttrValueMapUsingScalar(const OpAttrsRawEntry& entry,
Status ParseTfDataType(absl::string_view dtype, DataType* data_type) {
if (dtype == "DT_INT8") {
*data_type = DataType::DT_INT8;
return OkStatus();
return absl::OkStatus();
} else if (dtype == "DT_INT32") {
*data_type = DataType::DT_INT32;
return OkStatus();
return absl::OkStatus();
} else if (dtype == "DT_INT64") {
*data_type = DataType::DT_INT64;
return OkStatus();
return absl::OkStatus();
} else if (dtype == "DT_HALF") {
*data_type = DataType::DT_HALF;
return OkStatus();
return absl::OkStatus();
} else if (dtype == "DT_FLOAT") {
*data_type = DataType::DT_FLOAT;
return OkStatus();
return absl::OkStatus();
} else if (dtype == "DT_DOUBLE") {
*data_type = DataType::DT_DOUBLE;
return OkStatus();
return absl::OkStatus();
} else {
return errors::InvalidArgument("Unsupported dtype, ", std::string(dtype),
" in ParseTfDataType.");
Expand Down Expand Up @@ -432,10 +432,10 @@ tfrt::DType ConvertTfDataTypeToBefAttrType(DataType data_type) {
Status ParseBoolAttrValue(absl::string_view attr_value, bool* bool_val) {
if (attr_value == "false") {
*bool_val = false;
return OkStatus();
return absl::OkStatus();
} else if (attr_value == "true") {
*bool_val = true;
return OkStatus();
return absl::OkStatus();
} else {
return errors::InvalidArgument("Could not parse bool from \"", attr_value,
"\"");
Expand All @@ -448,7 +448,7 @@ Status ParseIntAttrValue(absl::string_view attr_value, int64_t* int_val) {
return errors::InvalidArgument("Could not parse int from \"", attr_value,
"\"");
}
return OkStatus();
return absl::OkStatus();
}

Status ParseTensorAttrValue(absl::string_view attr_value,
Expand All @@ -464,7 +464,7 @@ Status ParseTensorAttrValue(absl::string_view attr_value,
if (protobuf::TextFormat::ParseFromString(
static_cast<std::string>(attr_value), message) &&
tensor->FromProto(tensor_proto)) {
return OkStatus();
return absl::OkStatus();
} else {
return errors::InvalidArgument("Could not parse tensor value from \"",
attr_value, "\"");
Expand Down Expand Up @@ -498,7 +498,7 @@ Status ParseTensorShapeAttrValue(absl::string_view attr_value,
}
shape_val->push_back(int_val);
}
return OkStatus();
return absl::OkStatus();
}

bool IsUnusedAttribute(absl::string_view attr_name) {
Expand Down Expand Up @@ -576,14 +576,14 @@ Status SetUpScalarAttr(tfrt::TypedAttrBase bef_attr,
return tensorflow::errors::Internal("Failed to set up attribute.");
}

return OkStatus();
return absl::OkStatus();
}

Status SetUpScalarFunctionAttr(tfrt::StringAttr func_attr,
tensorflow::AttrValue& tf_attr) {
tfrt::string_view func_name = func_attr.GetValue();
tf_attr.mutable_func()->set_name(func_name.data(), func_name.size());
return OkStatus();
return absl::OkStatus();
}

void AddShapeToAttrList(tfrt::ShapeAttr shape,
Expand Down Expand Up @@ -618,7 +618,7 @@ Status SetUpListAttr(tfrt::AggregateAttr aggregate_attr,
return tensorflow::errors::Internal("Failed to set up list attr.");
}
}
return OkStatus();
return absl::OkStatus();
}

Status SetUpListAttr(tfrt::ArrayAttr array_attr,
Expand All @@ -627,7 +627,7 @@ Status SetUpListAttr(tfrt::ArrayAttr array_attr,

// Handle an empty array case.
if (array_attr.GetNumElements() == 0) {
return OkStatus();
return absl::OkStatus();
}

tfrt::BEFAttributeType element_type = array_attr.GetElementType();
Expand All @@ -638,19 +638,19 @@ Status SetUpListAttr(tfrt::ArrayAttr array_attr,
for (auto value : array_attr.GetValue<bool>()) {
list->add_b(value);
}
return OkStatus();
return absl::OkStatus();
}
case tfrt::DType::I64: {
for (auto value : array_attr.GetValue<int64_t>()) {
list->add_i(value);
}
return OkStatus();
return absl::OkStatus();
}
case tfrt::DType::F32: {
for (auto value : array_attr.GetValue<float>()) {
list->add_f(value);
}
return OkStatus();
return absl::OkStatus();
}
default:
return tensorflow::errors::Internal(
Expand All @@ -661,7 +661,7 @@ Status SetUpListAttr(tfrt::ArrayAttr array_attr,
for (auto value : array_attr.GetValue<tfrt::DType>()) {
list->add_type(ConvertBefAttrTypeToTfDataType(value));
}
return OkStatus();
return absl::OkStatus();
}

return tensorflow::errors::Internal("Failed to set up list attr.");
Expand Down Expand Up @@ -705,7 +705,7 @@ Status SetUpAttrValueMap(tfrt::AggregateAttr op_attr_array,
TF_RETURN_IF_ERROR(SetUpScalarFunctionAttr(attr_value, tf_attr));
}

return OkStatus();
return absl::OkStatus();
}

} // namespace tfd
Expand Down
12 changes: 6 additions & 6 deletions tensorflow/core/runtime_fallback/util/attr_util_test.cc
View file
Expand Up @@ -72,7 +72,7 @@ INSTANTIATE_TEST_SUITE_P(

TEST_P(ParseTfDataTypeTest, Ok) {
DataType data_type;
ASSERT_EQ(ParseTfDataType(GetParam().str_val, &data_type), OkStatus());
ASSERT_EQ(ParseTfDataType(GetParam().str_val, &data_type), absl::OkStatus());
EXPECT_EQ(data_type, GetParam().dtype);
}

Expand Down Expand Up @@ -210,7 +210,7 @@ TEST(UtilsTest, ParseTensorAttrValueOk) {
int_val: 1
int_val: 1
int_val: 1)pb";
ASSERT_EQ(ParseTensorAttrValue(tensor_str, &tensor), OkStatus());
ASSERT_EQ(ParseTensorAttrValue(tensor_str, &tensor), absl::OkStatus());
EXPECT_EQ(tensor.dtype(), DT_INT32);
EXPECT_EQ(tensor.NumElements(), 4);
}
Expand All @@ -225,7 +225,7 @@ TEST(UtilsTest, ParseTensorAttrValueReturnsInvalidArgument) {

TEST(UtilsTest, ParseTensorShapeAttrValueOk) {
std::vector<int64_t> dims;
ASSERT_THAT(ParseTensorShapeAttrValue("[1,2,3]", &dims), OkStatus());
ASSERT_THAT(ParseTensorShapeAttrValue("[1,2,3]", &dims), absl::OkStatus());
EXPECT_THAT(dims, ElementsAre(Eq(1), Eq(2), Eq(3)));
}

Expand All @@ -246,10 +246,10 @@ TEST(UtilsTest, ParseTensorShapeAttrValueInvalidArgumentEmptyString) {

TEST(UtilsTest, ParseBoolAttrValueOk) {
bool bool_val;
ASSERT_THAT(ParseBoolAttrValue("false", &bool_val), OkStatus());
ASSERT_THAT(ParseBoolAttrValue("false", &bool_val), absl::OkStatus());
EXPECT_FALSE(bool_val);

ASSERT_THAT(ParseBoolAttrValue("true", &bool_val), OkStatus());
ASSERT_THAT(ParseBoolAttrValue("true", &bool_val), absl::OkStatus());
EXPECT_TRUE(bool_val);
}

Expand All @@ -261,7 +261,7 @@ TEST(UtilsTest, ParseBoolAttrValueInvalidArgument) {

TEST(UtilsTest, ParseIntAttrValueOk) {
int64_t int_val;
ASSERT_THAT(ParseIntAttrValue("42", &int_val), OkStatus());
ASSERT_THAT(ParseIntAttrValue("42", &int_val), absl::OkStatus());
EXPECT_EQ(int_val, 42);
}

Expand Down

0 comments on commit d709dd2

Please sign in to comment.