Home
last modified time | relevance | path

Searched refs:tf_shared_lock (Results 1 – 25 of 50) sorted by relevance

12

/external/tensorflow/tensorflow/core/platform/default/
Dmutex.h102 class SCOPED_LOCKABLE tf_shared_lock {
106 explicit tf_shared_lock(mutex_type& mu) SHARED_LOCK_FUNCTION(mu) : mu_(&mu) { in tf_shared_lock() function
110 tf_shared_lock(mutex_type& mu, std::try_to_lock_t) SHARED_LOCK_FUNCTION(mu) in tf_shared_lock() function
119 tf_shared_lock(tf_shared_lock&& ml) noexcept SHARED_LOCK_FUNCTION(ml.mu_) in tf_shared_lock() function
123 ~tf_shared_lock() UNLOCK_FUNCTION() { in UNLOCK_FUNCTION()
137 #define tf_shared_lock(x) \ macro
/external/tensorflow/tensorflow/core/platform/
Dmutex_test.cc33 tf_shared_lock GetLock() { return tf_shared_lock{mu}; } in GetLock()
34 void Test() { tf_shared_lock lock = GetLock(); } in Test()
/external/tensorflow/tensorflow/stream_executor/platform/default/
Dmutex.h25 #undef tf_shared_lock
32 using tensorflow::tf_shared_lock;
35 #define tf_shared_lock(x) \ macro
/external/tensorflow/tensorflow/core/framework/
Dmodel.h141 tf_shared_lock l(mu_); in buffered_bytes()
147 tf_shared_lock l(mu_); in has_tunable_parameters()
159 tf_shared_lock l(mu_); in inputs()
168 tf_shared_lock l(mu_); in num_elements()
177 tf_shared_lock l(mu_); in processing_time()
217 tf_shared_lock l(mu_); in CollectTunableParameters()
230 tf_shared_lock l(mu_); in OutputTime()
237 tf_shared_lock l(mu_); in ProcessingTime()
248 tf_shared_lock l(mu_); in Snapshot()
Dmodel.cc397 tf_shared_lock l(mu_); in AddProcessingTime()
413 tf_shared_lock lock(mu_); in Optimize()
467 tf_shared_lock l(mu_); in RecordElement()
475 tf_shared_lock l(mu_); in RecordStart()
487 tf_shared_lock l(mu_); in RecordStop()
Dfunction_handle_cache.cc41 tf_shared_lock l(mu_); in Instantiate()
Dfunction.cc1030 tf_shared_lock l(other.mu_); in FunctionLibraryDefinition()
1055 tf_shared_lock l(mu_); in Contains()
1060 tf_shared_lock l(mu_); in Find()
1258 tf_shared_lock l(mu_); in FindGradient()
1268 tf_shared_lock l(mu_); in LookUp()
1278 tf_shared_lock l(mu_); in UniqueFunctionName()
1304 tf_shared_lock l(mu_); in GetAttrImpl()
1318 tf_shared_lock l(mu_); in ListFunctionNames()
1328 tf_shared_lock l(mu_); in ToProto()
Dresource_mgr.h444 tf_shared_lock l(mu_); in Lookup()
454 tf_shared_lock l(mu_); in LookupMany()
500 tf_shared_lock l(mu_);
/external/tensorflow/tensorflow/core/kernels/
Dtraining_op_helpers.h77 std::unique_ptr<std::vector<tf_shared_lock>> shared_locks) in VariableInputLockHolder()
101 std::unique_ptr<std::vector<tf_shared_lock>> shared_locks_;
172 std::unique_ptr<std::vector<tf_shared_lock>> shared_locks = in MaybeLockVariableInputMutexesInOrder()
173 absl::make_unique<std::vector<tf_shared_lock>>(); in MaybeLockVariableInputMutexesInOrder()
Dlookup_table_op.cc53 tf_shared_lock l(mu_); in size()
63 tf_shared_lock l(mu_); in Find()
108 tf_shared_lock l(mu_); in ExportValues()
138 tf_shared_lock l(mu_); in MemoryUsed()
170 tf_shared_lock l(mu_); in size()
181 tf_shared_lock l(mu_); in Find()
241 tf_shared_lock l(mu_); in ExportValues()
276 tf_shared_lock l(mu_); in MemoryUsed()
379 tf_shared_lock l(mu_); in size()
399 tf_shared_lock l(mu_); in Find()
[all …]
Dresource_variable_ops.cc145 tf_shared_lock ml(*variable->mu()); in Compute()
195 tf_shared_lock ml(*variables[i]->mu()); in Compute()
626 tf_shared_lock ml(*v->mu()); in Compute()
763 tf_shared_lock ml(*v->mu()); in Compute()
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dmodel_ops.cc75 tf_shared_lock l(*ensemble_resource->get_mutex()); in Compute()
94 tf_shared_lock l(*ensemble_resource->get_mutex()); in Compute()
153 tf_shared_lock l(*ensemble_resource->get_mutex()); in Compute()
Dprediction_ops.cc173 tf_shared_lock l(*ensemble_resource->get_mutex()); in Compute()
382 tf_shared_lock l(*ensemble_resource->get_mutex()); in Compute()
/external/tensorflow/tensorflow/stream_executor/
Dexecutor_cache.cc69 tf_shared_lock lock{mutex_}; in Get()
79 tf_shared_lock lock{entry->configurations_mutex}; in Get()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dparallel_interleave_dataset_op.cc642 tf_shared_lock l(ckpt_mu_); in WorkerThread()
666 tf_shared_lock l(ckpt_mu_); in WorkerThread()
688 tf_shared_lock ckpt_l(ckpt_mu_); in WorkerThread()
697 tf_shared_lock l(ckpt_mu_); in WorkerThread()
715 tf_shared_lock l(ckpt_mu_); in WorkerThread()
733 tf_shared_lock ckpt_l(ckpt_mu_); in WorkerThread()
748 tf_shared_lock ckpt_l(ckpt_mu_); in WorkerThread()
785 tf_shared_lock ckpt_l(ckpt_mu_); in WorkerThread()
Dstats_dataset_ops.cc105 tf_shared_lock l(mu_); in GetNextInternal()
215 tf_shared_lock l(mu_); in GetNextInternal()
Dignore_errors_dataset_op.cc89 tf_shared_lock l(mu_); in GetNextInternal()
/external/tensorflow/tensorflow/core/platform/cloud/
Dgcs_file_system.h146 tf_shared_lock l(block_cache_lock_); in block_size()
150 tf_shared_lock l(block_cache_lock_); in max_bytes()
154 tf_shared_lock l(block_cache_lock_); in max_staleness()
/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dresource_ops.cc84 tf_shared_lock l(*tree_ensemble_resource->get_mutex()); in Compute()
147 tf_shared_lock l(*tree_ensemble_resource->get_mutex()); in Compute()
/external/tensorflow/tensorflow/core/common_runtime/
Dexecutor_factory.cc63 tf_shared_lock l(executor_factory_lock); in GetFactory()
Dprocess_function_library_runtime.cc229 tf_shared_lock l(mu_); in GetHandle()
241 tf_shared_lock l(mu_); in GetHandleOnDevice()
261 tf_shared_lock l(mu_); in GetDeviceName()
271 tf_shared_lock l(mu_); in IsMultiDevice()
916 tf_shared_lock l(mu_); in Run()
927 tf_shared_lock l(mu_); in Run()
/external/tensorflow/tensorflow/core/kernels/data/
Dcache_dataset_ops.cc632 tf_shared_lock l(mu_); in IsClaimed()
638 tf_shared_lock l(mu_); in IsCompleted()
662 tf_shared_lock l(mu_); in at()
675 tf_shared_lock l(mu_); in size()
882 tf_shared_lock l(mu_); in Initialize()
Dunbounded_thread_pool.cc95 tf_shared_lock l(thread_pool_mu_); in size()
Dmulti_device_iterator_ops.cc95 tf_shared_lock l(mu_); in GetNextFromShard()
115 tf_shared_lock l(mu_); in function_library()
120 tf_shared_lock l(mu_); in lib()
Dfilter_by_component_dataset_op.cc119 tf_shared_lock l(mu_); in GetNextInternal()

12