Home
last modified time | relevance | path

Searched refs:SchedClosure (Results 1 – 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/core/framework/
Drendezvous_test.cc74 void SchedClosure(std::function<void()> fn) { in SchedClosure() function in tensorflow::__anon2f0f779b0111::LocalRendezvousTest
127 SchedClosure([this]() { in TEST_F()
140 SchedClosure([this]() { in TEST_F()
178 SchedClosure([this, i, micros]() { in TEST_F()
202 SchedClosure([this, i, micros, recv_done]() { in TEST_F()
222 SchedClosure([=]() { in TEST_F()
238 SchedClosure([this]() { in TEST_F()
253 SchedClosure([this]() { in TEST_F()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmaster.cc107 SchedClosure([this, sess]() { in GC()
357 SchedClosure([this, req, resp, done]() { in CreateSession()
502 SchedClosure([session, req, resp, done]() { in ExtendSession()
526 SchedClosure([session, req, resp, done]() { in PartialRunSetup()
548 SchedClosure([this, start_time, session, opts, req, resp, done]() { in RunStep()
581 SchedClosure([session, done]() { in CloseSession()
590 SchedClosure([this, req, resp, done]() { in ListDevices()
668 SchedClosure([sessions_to_close, done]() { in Reset()
692 SchedClosure(std::bind( in MakeCallable()
715 SchedClosure(std::bind( in RunCallable()
[all …]
Dworker_cache_partial.cc45 SchedClosure([this, &device_name, locality, done]() { in GetDeviceLocalityAsync()
Dcollective_rma_distributed.cc143 SchedClosure([s, done] { done(s); }); in RecvFromPeer()
Dcollective_rma_distributed_test.cc93 SchedClosure([this]() { in RecvBufAsync()
/external/tensorflow/tensorflow/core/common_runtime/
Dprocess_util.cc109 void SchedClosure(std::function<void()> closure) { in SchedClosure() function
111 return Env::Default()->SchedClosure(std::move(closure)); in SchedClosure()
116 Env::Default()->SchedClosure(std::bind( in SchedClosure()
Dcollective_param_resolver_local_test.cc175 Env::Default()->SchedClosure([this, i, cp, &note, &statuses]() { in TEST_F()
228 Env::Default()->SchedClosure([this, i, cp, &note, &statuses]() { in TEST_F()
266 Env::Default()->SchedClosure([this, i, cp, &note, &statuses]() { in TEST_F()
Dprocess_util.h52 void SchedClosure(std::function<void()> closure);
Dexecutor_test.cc335 SchedClosure([this]() { in TEST_F()
342 SchedClosure([this]() { in TEST_F()
349 SchedClosure([this]() { in TEST_F()
356 SchedClosure([this]() { in TEST_F()
Dbase_collective_executor.cc267 SchedClosure([col_impl, col_ctx, done_safe]() { in ExecuteAsync()
Ddirect_session.cc232 void DirectSession::SchedClosure(thread::ThreadPool* pool, in SchedClosure() function in tensorflow::DirectSession
635 SchedClosure(pool, std::move(c)); in RunInternal()
651 SchedClosure(device_thread_pool, std::move(c)); in RunInternal()
889 SchedClosure(pool, std::move(c)); in PRunSetup()
Ddirect_session.h334 void SchedClosure(thread::ThreadPool* pool, std::function<void()> c);
Dring_gatherer_test.cc246 SchedClosure([di, &done] { in Gather()
Dring_reducer_test.cc268 SchedClosure([di, &done] { in Reduce()
/external/tensorflow/tensorflow/core/platform/
Denv.h286 virtual void SchedClosure(std::function<void()> closure) = 0;
376 void SchedClosure(std::function<void()> closure) override { in SchedClosure() function
377 target_->SchedClosure(closure); in SchedClosure()
/external/tensorflow/tensorflow/core/platform/posix/
Denv.cc118 void SchedClosure(std::function<void()> closure) override { in SchedClosure() function in tensorflow::__anoncb12d2290111::PosixEnv
130 SchedClosure([this, micros, closure]() { in SchedClosureAfter()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Drpc_rendezvous_mgr_test.cc115 SchedClosure([this, rendez]() { in TEST_F()
129 SchedClosure([this, step_id]() { in TEST_F()
/external/tensorflow/tensorflow/core/util/
Dexec_on_stall.h47 env_->SchedClosure([this]() { in disabled_()
/external/tensorflow/tensorflow/core/platform/windows/
Denv.cc100 void SchedClosure(std::function<void()> closure) override { in SchedClosure() function in tensorflow::__anon839461010111::WindowsEnv
/external/tensorflow/tensorflow/cc/training/
Dcoordinator_test.cc45 Env::Default()->SchedClosure( in TEST()
Dqueue_runner_test.cc261 Env::Default()->SchedClosure( in TEST()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_execute_op.cc133 Env::Default()->SchedClosure([this, context, done]() { in ComputeAsync()
/external/tensorflow/tensorflow/core/grappler/clusters/
Dsingle_machine.cc331 Env::Default()->SchedClosure([this, n]() { in ShutdownSession()
/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_reducer_test.cc145 SchedClosure([di, &done, &done_mu, &done_cv] { in Reduce()
Dcollective_ops.cc45 c->env()->SchedClosure([this, c, done, col_exec]() { in CanProceedWithCompute()

12