Searched refs:input_ref_mutex (Results 1 – 12 of 12) sorted by relevance
42 OP_REQUIRES_OK(ctx, ctx->input_ref_mutex("counter", &mu)); in Compute()
64 mutex_lock l(*context->input_ref_mutex(0)); in Compute()
34 mutex_lock l(*context->input_ref_mutex(0)); in Compute()
87 mutex_lock l(*c->input_ref_mutex(0)); in Compute()166 mutex_lock l(*c->input_ref_mutex(0)); in Compute()
61 mutex_lock l(*context->input_ref_mutex(0)); in Compute()
126 return ctx->input_ref_mutex(input); in GetTrainingVariableMutex()
258 TF_RETURN_IF_ERROR(ctx->input_ref_mutex(input_name, &mu)); in GetTableHandle()
256 mutex_lock l(*c->input_ref_mutex(0)); in Compute()
387 Status OpKernelContext::input_ref_mutex(StringPiece name, mutex** out_mutex) { in input_ref_mutex() function in tensorflow::OpKernelContext395 *out_mutex = input_ref_mutex(start); in input_ref_mutex()418 tf_shared_lock l(*input_ref_mutex(index)); in mutable_input()434 mutex_lock l(*input_ref_mutex(index)); in replace_ref_input()588 mutex_lock l(*input_ref_mutex(index)); in delete_ref_input()610 tf_shared_lock l(*input_ref_mutex(start)); in mutable_input()
738 Status input_ref_mutex(StringPiece name, mutex** out_mutex);1181 mutex* input_ref_mutex(int index);1556 inline mutex* OpKernelContext::input_ref_mutex(int index) { in input_ref_mutex() function1643 return ctx_->input_ref_mutex(start_ + i); in ref_mutex()
536 TF_RETURN_IF_ERROR(ctx->input_ref_mutex(input_name, &mu));
42 mutex_lock l(*ctx->input_ref_mutex(0)); in Compute()