experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add LOCKS_EXCLUDED to a few FunctionLibraryDefinition methods

Browse Source 
PiperOrigin-RevId: 251565676
This commit is contained in:
Igor Ganichev 2019-06-04 19:41:29 -07:00 committed by TensorFlower Gardener
parent 8ac1075eac
commit a7dac9a9a0
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/core/framework/function.h
View File

@ -363,19 +363,20 @@ class FunctionLibraryDefinition : public OpRegistryInterface {
// a non-OK status if "func" was not found in the library, OK otherwise.
// Please be careful when replacing function: make sure all previous pointers
// returned by `Find()` are no longer in use.
Status ReplaceFunction(const string& func, const FunctionDef& fdef);
Status ReplaceFunction(const string& func, const FunctionDef& fdef)
LOCKS_EXCLUDED(mu_);
// Replaces the gradient corresponding to `grad.function_name()`. Returns
// a non-OK status if "grad.function_name()" was not found in the library, OK
// otherwise.
Status ReplaceGradient(const GradientDef& grad);
Status ReplaceGradient(const GradientDef& grad) LOCKS_EXCLUDED(mu_);
// Removes the function corresponding to 'func'. Returns a non-OK status if
// 'func' was not found in the library, OK otherwise.
// Please be careful when removing function: make sure there are no other
// nodes using the function, and all previous pointers returned by `Find()`
// are no longer in use.
Status RemoveFunction(const string& func);
Status RemoveFunction(const string& func) LOCKS_EXCLUDED(mu_);
// Adds the functions and gradients in 'other' to this function library.
// Duplicate functions and gradients are ignored.