diff options
| author |  Skye Wanderman-Milne <skyewm@google.com> | 2017-12-06 12:12:43 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-12-06 12:19:58 -0800 |
| commit | c7778898eaf001c82744a8f4c71eb9a880a158f0 (patch) | |
| tree | ec733467a50337a6945da45ec4f9a74cc67f8eaf /tensorflow/c/c_api.cc | |
| parent | 62ed39337eeb7d11a503454840d83af191f9b0ad (diff) | |
Uniquify names and prefixes in import_graph_def with C API enabled.
This makes the C API-enabled behavior the same as the current behavior.
PiperOrigin-RevId: 178133381
Diffstat (limited to 'tensorflow/c/c_api.cc')
| -rw-r--r-- | tensorflow/c/c_api.cc | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/tensorflow/c/c_api.cc b/tensorflow/c/c_api.cc index c8b4bfffd4..13253ced49 100644 --- a/tensorflow/c/c_api.cc +++ b/tensorflow/c/c_api.cc @@ -1850,6 +1850,16 @@ void TF_ImportGraphDefOptionsSetPrefix(TF_ImportGraphDefOptions* opts, opts->opts.prefix = prefix; } +void TF_ImportGraphDefOptionsSetUniquifyNames(TF_ImportGraphDefOptions* opts, + unsigned char uniquify_names) { + opts->opts.uniquify_names = uniquify_names; +} + +void TF_ImportGraphDefOptionsSetUniquifyPrefix(TF_ImportGraphDefOptions* opts, + unsigned char uniquify_prefix) { + opts->opts.uniquify_prefix = uniquify_prefix; +} + void TF_ImportGraphDefOptionsAddInputMapping(TF_ImportGraphDefOptions* opts, const char* src_name, int src_index, TF_Output dst) { |