diff options
author | karl@kubx.ca <karl@kubx.ca> | 2018-06-25 22:12:24 -0400 |
---|---|---|
committer | karl@kubx.ca <karl@kubx.ca> | 2018-07-25 21:10:29 -0400 |
commit | 2b303fddafec6b96a6868aaa76f55cc392b96586 (patch) | |
tree | 8b1da320c69ba5239f8bdd37bfac95cd02704d65 /tensorflow/c/c_api.h | |
parent | b24037513f12a5812a21b7ea92ff904ee9ea6cd8 (diff) |
Add scope name to TF_AddGradients
Diffstat (limited to 'tensorflow/c/c_api.h')
-rw-r--r-- | tensorflow/c/c_api.h | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/tensorflow/c/c_api.h b/tensorflow/c/c_api.h index c8ae6f2dd1..e896f68ce0 100644 --- a/tensorflow/c/c_api.h +++ b/tensorflow/c/c_api.h @@ -1138,12 +1138,16 @@ TF_CAPI_EXPORT extern void TF_AbortWhile(const TF_WhileParams* params); // shapes in `y`. // The partial derivatives are returned in `dy`. `dy` should be allocated to // size `nx`. +// `scope_name` names the scope (or sub-scope) into which all gradients +// operations are added. If `scope_name` is nullptr, "gradients" is used by +// default. // // WARNING: This function does not yet support all the gradients that python // supports. See // https://www.tensorflow.org/code/tensorflow/cc/gradients/README.md // for instructions on how to add C++ more gradients. -TF_CAPI_EXPORT void TF_AddGradients(TF_Graph* g, TF_Output* y, int ny, +TF_CAPI_EXPORT void TF_AddGradients(TF_Graph* g, const char* scope_name, + TF_Output* y, int ny, TF_Output* x, int nx, TF_Output* dx, TF_Status* status, TF_Output* dy); |