| /external/tensorflow/tensorflow/core/framework/ |
| D | rendezvous_test.cc | 76 void SchedClosure(std::function<void()> fn) { in SchedClosure() function in tensorflow::__anon3cca98520111::LocalRendezvousTest
129 SchedClosure([this]() { in TEST_F()
142 SchedClosure([this]() { in TEST_F()
175 SchedClosure([cm, &n]() { in TEST_F()
195 SchedClosure([this, cm, &n]() { in TEST_F()
214 SchedClosure([this, cm]() { in TEST_F()
300 SchedClosure([this, i, micros]() { in TEST_F()
324 SchedClosure([this, i, micros, recv_done]() { in TEST_F()
344 SchedClosure([=]() { in TEST_F()
360 SchedClosure([this]() { in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | master.cc | 107 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([session, req, resp, done = std::move(done)]() { in MakeCallable()
713 SchedClosure([session, opts, req, resp, done = std::move(done)]() { in RunCallable()
[all …]
|
| D | worker_cache_partial.cc | 45 SchedClosure([this, &device_name, locality, done]() { in GetDeviceLocalityAsync()
|
| D | collective_rma_distributed_test.cc | 104 SchedClosure([this]() { in RecvBufAsync()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_util.cc | 166 void SchedClosure(std::function<void()> closure) { in SchedClosure() function
168 return Env::Default()->SchedClosure(std::move(closure)); in SchedClosure()
173 Env::Default()->SchedClosure([id, closure = std::move(closure)]() { in SchedClosure()
|
| D | collective_param_resolver_local_test.cc | 180 Env::Default()->SchedClosure([this, i, cp, ¬e, &statuses]() { in TEST_F()
235 Env::Default()->SchedClosure([this, i, cp, ¬e, &statuses]() { in TEST_F()
277 Env::Default()->SchedClosure([this, i, cp, ¬e, &statuses]() { in TEST_F()
324 Env::Default()->SchedClosure([this, i, &cancel_mgr, &cp, &start, &done] { in TEST_F()
353 Env::Default()->SchedClosure([this, group_key, instance_key, i, in TEST_F()
373 Env::Default()->SchedClosure([this, group_key, instance_key, i, &cancel_mgr, in TEST_F()
403 Env::Default()->SchedClosure([this, group_key, instance_key, i, in TEST_F()
458 Env::Default()->SchedClosure( in TEST_F()
|
| D | process_util.h | 52 void SchedClosure(std::function<void()> closure);
|
| D | executor_test.cc | 347 SchedClosure([this]() { in TEST_F()
354 SchedClosure([this]() { in TEST_F()
361 SchedClosure([this]() { in TEST_F()
368 SchedClosure([this]() { in TEST_F()
|
| D | permuter_test.cc | 233 SchedClosure([di, &done] { in Permute()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | rpc_rendezvous_mgr_test.cc | 60 SchedClosure([done = std::move(done)]() { in RecvTensorAsync()
168 SchedClosure([this, rendez]() { in TEST_F()
182 SchedClosure([this, step_id]() { in TEST_F()
203 SchedClosure([this, cm, &n]() { in TEST_F()
227 SchedClosure([this, rendez, key, cm, &n]() { in TEST_F()
|
| D | grpc_worker_service.cc | 202 worker_->env()->env->SchedClosure(std::move(closure)); \
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/eager/ |
| D | grpc_eager_client_test.cc | 43 Env::Default()->SchedClosure([&client_cache, i, &counter]() { in TEST()
|
| D | grpc_eager_client.cc | 94 Env::Default()->SchedClosure([this]() { this->Unref(); }); in GrpcEagerClientThread()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | exec_on_stall.h | 47 env_->SchedClosure([this]() { in disabled_()
|
| /external/tensorflow/tensorflow/core/platform/default/ |
| D | env.cc | 170 void SchedClosure(std::function<void()> closure) override { in SchedClosure() function in tensorflow::__anone05a5c320111::PosixEnv
182 SchedClosure([this, micros, closure]() { in SchedClosureAfter()
|
| /external/tensorflow/tensorflow/core/platform/ |
| D | env.h | 431 virtual void SchedClosure(std::function<void()> closure) = 0;
522 void SchedClosure(std::function<void()> closure) override { in SchedClosure() function
523 target_->SchedClosure(closure); in SchedClosure()
|
| /external/tensorflow/tensorflow/core/platform/windows/ |
| D | env.cc | 127 void SchedClosure(std::function<void()> closure) override { in SchedClosure() function in tensorflow::__anon982373580111::WindowsEnv
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | tpu_compile_ops.cc | 137 env->SchedClosure([env, done]() { ExitCountdown(env, done); }); in Compute()
|
| D | xrt_execute_op.cc | 429 Env::Default()->SchedClosure([this, context, done]() { in ComputeAsync()
517 Env::Default()->SchedClosure([this, context, done]() { in ComputeAsync()
|
| /external/tensorflow/tensorflow/cc/training/ |
| D | coordinator_test.cc | 45 Env::Default()->SchedClosure( in TEST()
|
| D | queue_runner_test.cc | 261 Env::Default()->SchedClosure( in TEST()
|
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_event_mgr_test.cc | 459 Env::Default()->SchedClosure(runner); in BM_no_ops()
532 Env::Default()->SchedClosure( in BM_chain_ops()
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine.cc | 328 Env::Default()->SchedClosure([this, n]() { in ShutdownSession()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_test.cc | 173 SchedClosure([this, di, &done, &done_mu, &done_cv] { in RunCollective()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager.cc | 380 env->SchedClosure([this, nccl_stream]() { in GetCommunicator()
|