Home
last modified time | relevance | path

Searched refs:input_ref_mutex (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dassign_op.h64 mutex_lock l(*context->input_ref_mutex(0)); in Compute()
Dcount_up_to_op.cc35 mutex_lock l(*context->input_ref_mutex(0)); in Compute()
Ddense_update_ops.cc61 mutex_lock l(*context->input_ref_mutex(0)); in Compute()
Dscatter_op.cc81 mutex_lock l(*c->input_ref_mutex(0)); in Compute()
Dtraining_op_helpers.h127 return ctx->input_ref_mutex(input); in GetTrainingVariableMutex()
Dlookup_util.cc265 TF_RETURN_IF_ERROR(ctx->input_ref_mutex(input_name, &mu)); in GetTableHandle()
Dscatter_nd_op.cc263 mutex_lock l(*c->input_ref_mutex(0)); in Compute()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.cc412 Status OpKernelContext::input_ref_mutex(StringPiece name, mutex** out_mutex) { in input_ref_mutex() function in tensorflow::OpKernelContext
415 *out_mutex = input_ref_mutex(index); in input_ref_mutex()
436 tf_shared_lock l(*input_ref_mutex(index)); in mutable_input()
451 mutex_lock l(*input_ref_mutex(index)); in replace_ref_input()
590 mutex_lock l(*input_ref_mutex(index)); in delete_ref_input()
607 tf_shared_lock l(*input_ref_mutex(index)); in mutable_input()
Dop_kernel.h769 Status input_ref_mutex(StringPiece name, mutex** out_mutex);
1212 mutex* input_ref_mutex(int index);
1615 inline mutex* OpKernelContext::input_ref_mutex(int index) { in input_ref_mutex() function
1683 return ctx_->input_ref_mutex(start_ + i); in ref_mutex()
Dresource_mgr.h703 TF_RETURN_IF_ERROR(ctx->input_ref_mutex(input_name, &mu));