Searched refs:input_ref_mutex (Results 1 – 11 of 11) sorted by relevance
425 Status OpKernelContext::input_ref_mutex(StringPiece name, mutex** out_mutex) { in input_ref_mutex() function in tensorflow::OpKernelContext428 *out_mutex = input_ref_mutex(index); in input_ref_mutex()449 tf_shared_lock l(*input_ref_mutex(index)); in mutable_input()464 mutex_lock l(*input_ref_mutex(index)); in replace_ref_input()634 mutex_lock l(*input_ref_mutex(index)); in delete_ref_input()651 tf_shared_lock l(*input_ref_mutex(index)); in mutable_input()
750 Status input_ref_mutex(StringPiece name, mutex** out_mutex);1173 mutex* input_ref_mutex(int index);1586 inline mutex* OpKernelContext::input_ref_mutex(int index) { in input_ref_mutex() function1623 return ctx_->input_ref_mutex(start_ + i); in ref_mutex()
48 mutex_lock l(*context->input_ref_mutex(input_ref_index)); in AssignRefVariable()
748 TF_RETURN_IF_ERROR(ctx->input_ref_mutex(input_name, &mu));
35 mutex_lock l(*context->input_ref_mutex(0)); in Compute()
61 mutex_lock l(*context->input_ref_mutex(0)); in Compute()
91 mutex_lock l(*c->input_ref_mutex(0)); in Compute()
126 return ctx->input_ref_mutex(input); in GetTrainingVariableMutex()
271 TF_RETURN_IF_ERROR(ctx->input_ref_mutex(input_name, &mu)); in GetTableHandle()
264 mutex_lock l(*c->input_ref_mutex(0)); in Compute()
168 return cc_ctx->input_ref_mutex(input); in GetTrainingVariableMutex()