aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/c/c_api.cc
diff options
context:
space:
mode:
authorGravatar karl@kubx.ca <karl@kubx.ca>2018-07-08 00:21:45 -0400
committerGravatar karl@kubx.ca <karl@kubx.ca>2018-07-25 21:10:30 -0400
commit7ebdc9834bbc583bcc42551b660c8ed256ea7416 (patch)
treefd5d4d733834c102943917d6259cb31f67293120 /tensorflow/c/c_api.cc
parentab063cd57d7eda73bcbaf11d43f8b2e6708979a3 (diff)
1st code review: rename 'scope_name' to 'prefix', etc.
Diffstat (limited to 'tensorflow/c/c_api.cc')
-rw-r--r--tensorflow/c/c_api.cc9
1 files changed, 7 insertions, 2 deletions
diff --git a/tensorflow/c/c_api.cc b/tensorflow/c/c_api.cc
index 96653154e5..32b0b70620 100644
--- a/tensorflow/c/c_api.cc
+++ b/tensorflow/c/c_api.cc
@@ -2387,7 +2387,12 @@ void TF_FinishWhile(const TF_WhileParams* params, TF_Status* status,
void TF_AbortWhile(const TF_WhileParams* params) { FreeWhileResources(params); }
-void TF_AddGradients(TF_Graph* g, const char* scope_name, TF_Output* y,
+void TF_AddGradients(TF_Graph* g, TF_Output* y, int ny, TF_Output* x, int nx,
+ TF_Output* dx, TF_Status* status, TF_Output* dy) {
+ TF_AddGradientsWithPrefix(g, nullptr, y, ny, x, nx, dx, status, dy);
+}
+
+void TF_AddGradientsWithPrefix(TF_Graph* g, const char* prefix, TF_Output* y,
int ny, TF_Output* x, int nx, TF_Output* dx,
TF_Status* status, TF_Output* dy) {
#ifdef __ANDROID__
@@ -2408,7 +2413,7 @@ void TF_AddGradients(TF_Graph* g, const char* scope_name, TF_Output* y,
tensorflow::Scope scope =
NewInternalScope(&g->graph, &status->status, &g->refiner)
- .NewSubScope(scope_name != nullptr ? scope_name : "gradients");
+ .NewSubScope(prefix != nullptr ? prefix : "gradients");
if (dx != nullptr) {
std::vector<tensorflow::Output> dx_arg = OutputsFromTFOutputs(dx, ny);