Searched refs:GetTrainingVariableMutex (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | training_op_helpers.h | 110 mutex* GetTrainingVariableMutex(OpKernelContext* ctx, int input, bool sparse, in GetTrainingVariableMutex() function
159 GetTrainingVariableMutex<Device, T>(ctx, input, sparse, &var); in MaybeLockVariableInputMutexesInOrder()
178 mutex* mu = GetTrainingVariableMutex<Device, T>(ctx, input, sparse, &var); in MaybeLockVariableInputMutexesInOrder()
|
| D | training_ops.cc | 606 mutex* mu = GetTrainingVariableMutex<Device, T>(ctx, 0, sparse, &resource); in Compute()
762 mutex* mu = GetTrainingVariableMutex<CPUDevice, T>(ctx, 0, sparse, &var); in Compute()
|