Searched refs:GetTrainingVariableMutex (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | training_op_helpers.h | 111 mutex* GetTrainingVariableMutex(OpKernelContext* ctx, int input, bool sparse, in GetTrainingVariableMutex() function
160 GetTrainingVariableMutex<Device, T>(ctx, input, sparse, &var); in MaybeLockVariableInputMutexesInOrder()
177 mutex* mu = GetTrainingVariableMutex<Device, T>(ctx, input, sparse, &var); in MaybeLockVariableInputMutexesInOrder()
|