Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/common_runtime/data/
Dstandalone.cc50 std::unique_ptr<DeviceMgr> device_mgr = MakeUnique<DeviceMgr>( in FromGraph() local
52 Device* device = device_mgr->ListDevices()[0]; in FromGraph()
59 device_mgr.get(), Env::Default(), TF_GRAPH_DEF_VERSION, in FromGraph()
79 WrapUnique(new Dataset(dataset, device_mgr.release(), pflr.release(), in FromGraph()
111 Dataset::Dataset(DatasetBase* dataset, DeviceMgr* device_mgr, in Dataset() argument
115 device_mgr_(device_mgr), in Dataset()
Dstandalone.h105 Dataset(DatasetBase* dataset, DeviceMgr* device_mgr,
/external/tensorflow/tensorflow/core/distributed_runtime/
Dsession_mgr.cc41 worker_env->device_mgr, in SessionMgr()
43 new GraphMgr(worker_env, worker_env->device_mgr)))), in SessionMgr()
87 auto device_mgr = MakeUnique<DeviceMgr>(std::move(renamed_devices)); in CreateSession() local
88 auto graph_mgr = MakeUnique<GraphMgr>(worker_env_, device_mgr.get()); in CreateSession()
92 std::move(device_mgr), std::move(graph_mgr))); in CreateSession()
97 auto graph_mgr = MakeUnique<GraphMgr>(worker_env_, worker_env_->device_mgr); in CreateSession()
101 worker_env_->device_mgr, std::move(graph_mgr)); in CreateSession()
Dsession_mgr_test.cc52 env_.device_mgr = device_mgr_.get(); in SessionMgrTest()
115 std::vector<Device*> devices_1 = session_1->device_mgr()->ListDevices(); in TEST_F()
121 std::vector<Device*> devices_2 = session_2->device_mgr()->ListDevices(); in TEST_F()
127 std::vector<Device*> devices_3 = session_3->device_mgr()->ListDevices(); in TEST_F()
133 std::vector<Device*> devices_4 = session_4->device_mgr()->ListDevices(); in TEST_F()
Dworker_session.h48 DeviceMgr* device_mgr() { in device_mgr() function
62 std::unique_ptr<DeviceMgr> device_mgr,
Dworker_session.cc98 std::unique_ptr<DeviceMgr> device_mgr, in WorkerSession() argument
106 device_mgr_(std::move(device_mgr)), in WorkerSession()
Dworker_env.h56 DeviceMgr* device_mgr = nullptr; member
Dworker.cc35 DeviceMgr* dm = env_->device_mgr; in GetStatusAsync()
368 env_->device_mgr->ClearContainers(containers); in CleanupAllAsync()
435 TF_RETURN_IF_ERROR(env_->device_mgr->LookupDevice(local_name, src_dev)); in PrepareRecvTensor()
/external/tensorflow/tensorflow/core/grappler/clusters/
Dsingle_machine.cc224 const DeviceMgr* device_mgr; in GetPeakMemoryUsage() local
225 TF_RETURN_IF_ERROR(session_->LocalDeviceManager(&device_mgr)); in GetPeakMemoryUsage()
226 std::vector<Device*> devices = device_mgr->ListDevices(); in GetPeakMemoryUsage()
373 const DeviceMgr* device_mgr; in ResetSession() local
374 TF_RETURN_IF_ERROR(session_->LocalDeviceManager(&device_mgr)); in ResetSession()
375 for (auto d : device_mgr->ListDevices()) { in ResetSession()
453 const DeviceMgr* device_mgr; in ClearAllocatorStats() local
454 TF_RETURN_IF_ERROR(session_->LocalDeviceManager(&device_mgr)); in ClearAllocatorStats()
455 std::vector<Device*> devices = device_mgr->ListDevices(); in ClearAllocatorStats()
/external/tensorflow/tensorflow/lite/delegates/flex/
Ddelegate_data.cc38 std::unique_ptr<tensorflow::DeviceMgr> device_mgr = in Prepare() local
42 new tensorflow::IntraProcessRendezvous(device_mgr.get()); in Prepare()
46 /*async=*/false, std::move(device_mgr), rendezvous)); in Prepare()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcontext.cc49 std::unique_ptr<const DeviceMgr> device_mgr, in EagerContext() argument
51 : EagerContext(opts, default_policy, async, device_mgr.release(), in EagerContext()
56 bool async, const DeviceMgr* device_mgr, in EagerContext() argument
59 devices_(device_mgr->ListDevices()), in EagerContext()
63 device_mgr, opts.env, TF_GRAPH_DEF_VERSION, &func_lib_def_, in EagerContext()
74 local_device_manager_.reset(device_mgr); in EagerContext()
77 local_unowned_device_manager_ = device_mgr; in EagerContext()
430 std::unique_ptr<ServerInterface> server, DeviceMgr* device_mgr, in StoreCollectiveOpsServer() argument
436 local_unowned_device_manager_ = device_mgr; in StoreCollectiveOpsServer()
Dcontext.h81 std::unique_ptr<const DeviceMgr> device_mgr,
86 const DeviceMgr* device_mgr, bool device_mgr_owned,
231 std::unique_ptr<ServerInterface> server, DeviceMgr* device_mgr,
/external/tensorflow/tensorflow/contrib/gdr/
Dgdr_server_lib.cc68 env->device_mgr->ListDevices()[0]->name(), &default_worker_name, in Init()
72 new DeviceResolverDistributed(env->device_mgr, worker_cache, in Init()
76 config, env->device_mgr, dev_resolver.get(), worker_cache, in Init()
79 config, env->device_mgr, std::move(dev_resolver), in Init()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_server_lib.cc104 delete worker_env_.device_mgr; in ~GrpcServer()
160 worker_env_.device_mgr = new DeviceMgr(std::move(devices)); in Init()
161 master_env_.local_devices = worker_env_.device_mgr->ListDevices(); in Init()
162 worker_env_.local_devices = worker_env_.device_mgr->ListDevices(); in Init()
230 new DeviceResolverDistributed(worker_env_.device_mgr, worker_cache, in Init()
233 new CollectiveParamResolverDistributed(config, worker_env_.device_mgr, in Init()
237 config, worker_env_.device_mgr, std::move(dev_resolver), in Init()
/external/tensorflow/tensorflow/c/eager/
Dc_api.cc83 std::unique_ptr<tensorflow::DeviceMgr>* device_mgr) { in GetAllRemoteDevices() argument
109 *device_mgr = std::move(remote_device_mgr); in GetAllRemoteDevices()
227 auto* device_mgr = grpc_server->worker_env()->device_mgr; in UpdateTFE_ContextWithServerDef() local
231 std::move(remote_device_mgr), remote_contexts, r, device_mgr, in UpdateTFE_ContextWithServerDef()
342 std::unique_ptr<tensorflow::DeviceMgr> device_mgr( in TFE_NewContext() local
346 new tensorflow::IntraProcessRendezvous(device_mgr.get()); in TFE_NewContext()
349 opts->async, device_mgr.release(), in TFE_NewContext()
355 const tensorflow::DeviceMgr* device_mgr = nullptr; in TFE_NewContextFromSession() local
356 status->status = sess->session->LocalDeviceManager(&device_mgr); in TFE_NewContextFromSession()
359 new tensorflow::IntraProcessRendezvous(device_mgr); in TFE_NewContextFromSession()
[all …]
Dc_api_internal.h68 const tensorflow::DeviceMgr* device_mgr, bool device_mgr_owned, in TFE_Context()
73 async, device_mgr, device_mgr_owned, rendezvous) {} in TFE_Context()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph_test.cc192 const DeviceMgr* device_mgr = nullptr; in TEST() local
193 TF_ASSERT_OK(session->LocalDeviceManager(&device_mgr)); in TEST()
194 for (auto d : device_mgr->ListDevices()) { in TEST()
/external/tensorflow/tensorflow/core/common_runtime/
Dprocess_function_library_runtime.cc59 const DeviceMgr* device_mgr, Env* env, int graph_def_version, in ProcessFunctionLibraryRuntime() argument
65 device_mgr_(device_mgr), in ProcessFunctionLibraryRuntime()
70 if (device_mgr == nullptr) { in ProcessFunctionLibraryRuntime()
76 for (Device* d : device_mgr->ListDevices()) { in ProcessFunctionLibraryRuntime()
78 device_mgr, env, d, graph_def_version, lib_def_, default_thread_pool, in ProcessFunctionLibraryRuntime()
84 const DeviceMgr* device_mgr, Env* env, int graph_def_version, in ProcessFunctionLibraryRuntime() argument
91 device_mgr_(device_mgr), in ProcessFunctionLibraryRuntime()
96 if (device_mgr == nullptr) { in ProcessFunctionLibraryRuntime()
102 for (Device* d : device_mgr->ListDevices()) { in ProcessFunctionLibraryRuntime()
104 device_mgr, env, d, graph_def_version, lib_def_, default_thread_pool, in ProcessFunctionLibraryRuntime()
Dfunction.h57 const DeviceMgr* device_mgr, Env* env, Device* device,
67 const DeviceMgr* device_mgr, Env* env, Device* device,
Dprocess_function_library_runtime.h35 const DeviceMgr* device_mgr, Env* env, int graph_def_version,
42 ProcessFunctionLibraryRuntime(const DeviceMgr* device_mgr, Env* env,
140 const DeviceMgr* device_mgr() { return device_mgr_; } in device_mgr() function
Drendezvous_mgr.cc34 IntraProcessRendezvous::IntraProcessRendezvous(const DeviceMgr* device_mgr) in IntraProcessRendezvous() argument
35 : device_mgr_(device_mgr), local_(NewLocalRendezvous()) {} in IntraProcessRendezvous()
Drendezvous_mgr.h42 explicit IntraProcessRendezvous(const DeviceMgr* device_mgr);
/external/tensorflow/tensorflow/core/kernels/data/
Diterator_ops.cc59 std::unique_ptr<DeviceMgr> device_mgr, in IteratorResource() argument
64 device_mgr_(std::move(device_mgr)), in IteratorResource()
422 std::unique_ptr<DeviceMgr> device_mgr(nullptr); in Compute() local
430 lib = CreatePrivateFLR(context, &device_mgr, &flib_def, &pflr); in Compute()
444 [context, lib, &device_mgr, &flib_def, &pflr, in Compute()
448 graph_def_version_, std::move(device_mgr), in Compute()
477 OpKernelContext* ctx, std::unique_ptr<DeviceMgr>* device_mgr, in CreatePrivateFLR() argument
484 *device_mgr = absl::make_unique<DeviceMgr>(RenamedDevice::NewRenamedDevice( in CreatePrivateFLR()
490 device_mgr->get(), ctx->env(), graph_def_version_, flib_def->get(), in CreatePrivateFLR()
510 std::unique_ptr<DeviceMgr> device_mgr(nullptr); in Compute() local
[all …]
/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/core/kernels/
Dpartitioned_function_ops.cc186 TF_RETURN_IF_ERROR(lib->device_mgr()->LookupDevice("CPU:0", &cpu_device)); in Instantiate()
228 Rendezvous* rendez = new IntraProcessRendezvous(lib->device_mgr()); in RunFunction()

12