diff options
| author |  Olivia Nordquist <nolivia@google.com> | 2017-10-31 16:43:46 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-10-31 16:47:44 -0700 |
| commit | f97e7c69b84dac8c3c8c78204d48816036b9bead (patch) | |
| tree | 67646716fe3074bd9afeb25f5170d24e84e24d87 /tensorflow/c/python_api.h | |
| parent | 8e732a31246f52cd277536e1cb9bab7aa1807d60 (diff) | |
partially exposing the _set_attr and _get_attr method in python
PiperOrigin-RevId: 174113043
Diffstat (limited to 'tensorflow/c/python_api.h')
| -rw-r--r-- | tensorflow/c/python_api.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/c/python_api.h b/tensorflow/c/python_api.h index ab71a4170b..f54585b0a1 100644 --- a/tensorflow/c/python_api.h +++ b/tensorflow/c/python_api.h @@ -25,6 +25,11 @@ namespace tensorflow { void AddControlInput(TF_Graph* graph, TF_Operation* op, TF_Operation* input); +// Changes an attr value in the node_def Protocol Buffer and sets a status upon +// completion. +void SetAttr(TF_Graph* graph, TF_Operation* op, const char* attr_name, + TF_Buffer* attr_value_proto, TF_Status* status); + void SetRequestedDevice(TF_Graph* graph, TF_Operation* op, const char* device); void UpdateEdge(TF_Graph* graph, TF_Output new_src, TF_Input dst, |