experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix in c_api DEFINE_GETATTR

Browse Source 
PiperOrigin-RevId: 314629630
Change-Id: Iad90c91949e834b710a05136a86c689c93012fa4
This commit is contained in:
Yanhua Sun 2020-06-03 16:45:00 -07:00 committed by TensorFlower Gardener
parent c0b6b669e2
commit 10acce5990
2 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/c/c_api.cc
View File

@ -1386,6 +1386,7 @@ void TF_OperationGetAttrStringList(TF_Operation* oper, const char* attr_name,
cpp_type v; \ cpp_type v; \
status->status = \ status->status = \
tensorflow::GetNodeAttr(oper->node.attrs(), attr_name, &v); \ tensorflow::GetNodeAttr(oper->node.attrs(), attr_name, &v); \
if (!status->status.ok()) return; \
*value = static_cast<c_type>(v); \ *value = static_cast<c_type>(v); \
} \ } \
void func##List(TF_Operation* oper, const char* attr_name, c_type* values, \ void func##List(TF_Operation* oper, const char* attr_name, c_type* values, \

3
tensorflow/python/BUILD
View File

@ -4877,9 +4877,6 @@ cuda_py_test(
srcs = ["ops/control_flow_ops_test.py"], srcs = ["ops/control_flow_ops_test.py"],
python_version = "PY3", python_version = "PY3",
shard_count = 2, shard_count = 2,
tags = [
"noasan", # b/158098804
],
deps = [ deps = [
":array_ops", ":array_ops",
":cond_v2", ":cond_v2",