Home
last modified time | relevance | path

Searched refs:device_mgr (Results 1 – 25 of 33) sorted by relevance

12

/external/tensorflow/tensorflow/core/common_runtime/
Dprocess_function_library_runtime.cc29 const DeviceMgr* device_mgr, Env* env, int graph_def_version, in ProcessFunctionLibraryRuntime() argument
33 : device_mgr_(device_mgr), in ProcessFunctionLibraryRuntime()
37 if (device_mgr == nullptr) { in ProcessFunctionLibraryRuntime()
43 for (Device* d : device_mgr->ListDevices()) { in ProcessFunctionLibraryRuntime()
45 NewFunctionLibraryRuntime(device_mgr, env, d, graph_def_version, in ProcessFunctionLibraryRuntime()
51 const DeviceMgr* device_mgr, Env* env, int graph_def_version, in ProcessFunctionLibraryRuntime() argument
56 : device_mgr_(device_mgr), in ProcessFunctionLibraryRuntime()
60 if (device_mgr == nullptr) { in ProcessFunctionLibraryRuntime()
66 for (Device* d : device_mgr->ListDevices()) { in ProcessFunctionLibraryRuntime()
68 device_mgr, env, d, graph_def_version, lib_def, optimizer_options, in ProcessFunctionLibraryRuntime()
Dfunction.h56 const DeviceMgr* device_mgr, Env* env, Device* device,
66 const DeviceMgr* device_mgr, Env* env, Device* device,
Drendezvous_mgr.cc34 IntraProcessRendezvous::IntraProcessRendezvous(const DeviceMgr* device_mgr) in IntraProcessRendezvous() argument
35 : device_mgr_(device_mgr), local_(NewLocalRendezvous()) {} in IntraProcessRendezvous()
Dprocess_function_library_runtime.h33 const DeviceMgr* device_mgr, Env* env, int graph_def_version,
39 ProcessFunctionLibraryRuntime(const DeviceMgr* device_mgr, Env* env,
Drendezvous_mgr.h42 explicit IntraProcessRendezvous(const DeviceMgr* device_mgr);
Dfunction.cc945 const DeviceMgr* device_mgr, Env* env, Device* device, in NewFunctionLibraryRuntime() argument
951 device_mgr, env, device, graph_def_version, lib_def, optimizer_options, in NewFunctionLibraryRuntime()
956 const DeviceMgr* device_mgr, Env* env, Device* device, in NewFunctionLibraryRuntime() argument
960 return NewFunctionLibraryRuntime(device_mgr, env, device, graph_def_version, in NewFunctionLibraryRuntime()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dsession_mgr.cc40 std::unique_ptr<DeviceMgr>(worker_env->device_mgr), in SessionMgr()
42 new GraphMgr(worker_env, worker_env->device_mgr)))), in SessionMgr()
79 std::unique_ptr<DeviceMgr> device_mgr(new DeviceMgr(renamed_devices)); in CreateSession() local
82 new GraphMgr(worker_env_, device_mgr.get())); in CreateSession()
86 std::move(device_mgr), std::move(graph_mgr))); in CreateSession()
Dsession_mgr_test.cc85 std::vector<Device*> devices_1 = session_1->device_mgr->ListDevices(); in TEST_F()
90 std::vector<Device*> devices_2 = session_2->device_mgr->ListDevices(); in TEST_F()
95 std::vector<Device*> devices_3 = session_3->device_mgr->ListDevices(); in TEST_F()
100 std::vector<Device*> devices_4 = session_4->device_mgr->ListDevices(); in TEST_F()
Dworker_session.cc93 std::unique_ptr<DeviceMgr> device_mgr, in WorkerSession() argument
98 device_mgr(std::move(device_mgr)), in WorkerSession()
Dworker_session.h46 const std::unique_ptr<DeviceMgr> device_mgr; member
58 std::unique_ptr<DeviceMgr> device_mgr,
Dworker.cc33 DeviceMgr* dm = env_->device_mgr; in GetStatusAsync()
297 env_->device_mgr->ClearContainers(containers); in CleanupAllAsync()
317 TF_RETURN_IF_ERROR(env_->device_mgr->LookupDevice(local_name, src_dev)); in PrepareRecvTensor()
Dworker_env.h55 DeviceMgr* device_mgr = nullptr; member
Dbase_rendezvous_mgr.cc256 Status s = sess->device_mgr->LookupDevice(parsed.src_device, &src_device); in SameWorkerRecvDone()
262 s = sess->device_mgr->LookupDevice(parsed.dst_device, &dst_device); in SameWorkerRecvDone()
Dgraph_mgr.h71 explicit GraphMgr(const WorkerEnv* worker_env, DeviceMgr* device_mgr);
Dgraph_mgr.cc52 GraphMgr::GraphMgr(const WorkerEnv* worker_env, DeviceMgr* device_mgr) in GraphMgr() argument
53 : worker_env_(worker_env), device_mgr_(device_mgr), table_(5) { in GraphMgr()
/external/tensorflow/tensorflow/core/grappler/clusters/
Dsingle_machine.cc211 const DeviceMgr* device_mgr; in GetPeakMemoryUsage() local
212 TF_RETURN_IF_ERROR(session_->LocalDeviceManager(&device_mgr)); in GetPeakMemoryUsage()
213 std::vector<Device*> devices = device_mgr->ListDevices(); in GetPeakMemoryUsage()
400 const DeviceMgr* device_mgr; in ClearAllocatorStats() local
401 TF_RETURN_IF_ERROR(session_->LocalDeviceManager(&device_mgr)); in ClearAllocatorStats()
402 std::vector<Device*> devices = device_mgr->ListDevices(); in ClearAllocatorStats()
/external/tensorflow/tensorflow/c/eager/
Dc_api_internal.h50 rendezvous(new tensorflow::IntraProcessRendezvous(s->device_mgr)), in TFE_Context()
52 session->device_mgr, opts.session_options.options.env, in TFE_Context()
Dc_api.cc90 if (session->device_mgr == nullptr || session->devices.empty()) { in TFE_NewContext()
192 status->status = ctx->session->device_mgr->LookupDevice(device_name, &dstd); in TFE_TensorHandleCopyToDevice()
278 op->ctx->session->device_mgr->LookupDevice(device_name, &d); in TFE_OpSetDevice()
/external/tensorflow/tensorflow/core/kernels/data/
Diterator_ops.cc87 std::unique_ptr<DeviceMgr> device_mgr, in IteratorResource() argument
91 : device_mgr_(std::move(device_mgr)), in IteratorResource()
467 std::unique_ptr<DeviceMgr> device_mgr(nullptr); in Compute() local
475 lib = CreatePrivateFLR(context, &device_mgr, &flib_def, &pflr); in Compute()
489 [lib, &device_mgr, &flib_def, &pflr, in Compute()
493 std::move(device_mgr), std::move(flib_def), in Compute()
542 OpKernelContext* ctx, std::unique_ptr<DeviceMgr>* device_mgr, in CreatePrivateFLR() argument
552 device_mgr->reset(new DeviceMgr({wrapped_device})); in CreatePrivateFLR()
556 device_mgr->get(), ctx->env(), graph_def_version_, flib_def->get(), in CreatePrivateFLR()
/external/tensorflow/tensorflow/contrib/mpi/
Dmpi_rendezvous_mgr.cc77 s = env_->device_mgr->LookupDevice(parsed.dst_device, &dst_device); in RecvFromRemoteAsync()
110 s = env_->device_mgr->LookupDevice(parsed.dst_device, &dst_device); in RecvFromRemoteAsync()
217 Status s = this->worker_env_2->device_mgr->LookupDevice( in AddRequest()
/external/tensorflow/tensorflow/contrib/verbs/
Drdma_rendezvous_mgr.cc70 s = env_->device_mgr->LookupDevice(parsed.dst_device, &dst_dev); in RecvFromRemoteAsync()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_server_lib.cc97 delete worker_env_.device_mgr; in ~GrpcServer()
126 worker_env_.device_mgr = new DeviceMgr(worker_env_.local_devices); in Init()
Drpc_rendezvous_mgr.cc230 s = sess->device_mgr->LookupDevice(parsed.dst_device, &dst_device); in RecvFromRemoteAsync()
/external/tensorflow/tensorflow/c/
Dc_api_internal.h134 const tensorflow::DeviceMgr* device_mgr; // Owned by session. member
/external/tensorflow/tensorflow/contrib/gdr/
Dgdr_rendezvous_mgr.cc158 Status s = sess->device_mgr->LookupDevice(parsed.dst_device, &dst_device); in RecvFromRemoteAsync()

12