Fix in c_api DEFINE_GETATTR

PiperOrigin-RevId: 314629630 Change-Id: Iad90c91949e834b710a05136a86c689c93012fa4
2020-06-03 16:45:00 -07:00 · 2020-06-03 16:45:00 -07:00 · 10acce5990
commit 10acce5990
parent c0b6b669e2
2 changed files with 1 additions and 3 deletions
--- a/tensorflow/c/c_api.cc
+++ b/tensorflow/c/c_api.cc
@ -1386,6 +1386,7 @@ void TF_OperationGetAttrStringList(TF_Operation* oper, const char* attr_name,
    cpp_type v;                                                              \
    status->status =                                                         \
        tensorflow::GetNodeAttr(oper->node.attrs(), attr_name, &v);          \
+    if (!status->status.ok()) return;                                        \
    *value = static_cast<c_type>(v);                                         \
  }                                                                          \
  void func##List(TF_Operation* oper, const char* attr_name, c_type* values, \
--- a/tensorflow/python/BUILD
+++ b/tensorflow/python/BUILD
@ -4877,9 +4877,6 @@ cuda_py_test(
    srcs = ["ops/control_flow_ops_test.py"],
    python_version = "PY3",
    shard_count = 2,
-    tags = [
-        "noasan",  # b/158098804
-    ],
    deps = [
        ":array_ops",
        ":cond_v2",