diff options
author | 2017-03-10 14:43:23 -0800 | |
---|---|---|
committer | 2017-03-10 15:18:15 -0800 | |
commit | 0386a01ad3beb28364599d82199be1c0837b3fa9 (patch) | |
tree | 3a1d2ef947a7bf37286efc0e8ff760e0401ab319 /tensorflow/c/c_api.h | |
parent | e73ceaebb209a1e577e7240fba41c692c89143d0 (diff) |
Merge changes from github.
Change: 149800363
Diffstat (limited to 'tensorflow/c/c_api.h')
-rw-r--r-- | tensorflow/c/c_api.h | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/tensorflow/c/c_api.h b/tensorflow/c/c_api.h index 02ba0ac509..f837b68d76 100644 --- a/tensorflow/c/c_api.h +++ b/tensorflow/c/c_api.h @@ -810,6 +810,13 @@ extern void TF_ImportGraphDefOptionsAddInputMapping( TF_ImportGraphDefOptions* opts, const char* src_name, int src_index, TF_Output dst); +// Set any imported nodes with control input `src_name` to have that input +// replaced with `dst`. `src_name` refers to a node in the graph to be imported, +// `dst` references an operation already existing in the graph being imported +// into. +extern void TF_GraphImportGraphDefOptionsRemapControlDependency( + TF_ImportGraphDefOptions* opts, const char* src_name, TF_Operation* dst); + // Cause the imported graph to have a control dependency on `oper`. `oper` // should exist in the graph being imported into. extern void TF_ImportGraphDefOptionsAddControlDependency( |