/external/tensorflow/tensorflow/core/common_runtime/ |
D | process_function_library_runtime.cc | 29 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()
|
D | function.h | 56 const DeviceMgr* device_mgr, Env* env, Device* device, 66 const DeviceMgr* device_mgr, Env* env, Device* device,
|
D | rendezvous_mgr.cc | 34 IntraProcessRendezvous::IntraProcessRendezvous(const DeviceMgr* device_mgr) in IntraProcessRendezvous() argument 35 : device_mgr_(device_mgr), local_(NewLocalRendezvous()) {} in IntraProcessRendezvous()
|
D | process_function_library_runtime.h | 33 const DeviceMgr* device_mgr, Env* env, int graph_def_version, 39 ProcessFunctionLibraryRuntime(const DeviceMgr* device_mgr, Env* env,
|
D | rendezvous_mgr.h | 42 explicit IntraProcessRendezvous(const DeviceMgr* device_mgr);
|
D | function.cc | 945 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/ |
D | session_mgr.cc | 40 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()
|
D | session_mgr_test.cc | 85 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()
|
D | worker_session.cc | 93 std::unique_ptr<DeviceMgr> device_mgr, in WorkerSession() argument 98 device_mgr(std::move(device_mgr)), in WorkerSession()
|
D | worker_session.h | 46 const std::unique_ptr<DeviceMgr> device_mgr; member 58 std::unique_ptr<DeviceMgr> device_mgr,
|
D | worker.cc | 33 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()
|
D | worker_env.h | 55 DeviceMgr* device_mgr = nullptr; member
|
D | base_rendezvous_mgr.cc | 256 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()
|
D | graph_mgr.h | 71 explicit GraphMgr(const WorkerEnv* worker_env, DeviceMgr* device_mgr);
|
D | graph_mgr.cc | 52 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/ |
D | single_machine.cc | 211 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/ |
D | c_api_internal.h | 50 rendezvous(new tensorflow::IntraProcessRendezvous(s->device_mgr)), in TFE_Context() 52 session->device_mgr, opts.session_options.options.env, in TFE_Context()
|
D | c_api.cc | 90 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/ |
D | iterator_ops.cc | 87 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/ |
D | mpi_rendezvous_mgr.cc | 77 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/ |
D | rdma_rendezvous_mgr.cc | 70 s = env_->device_mgr->LookupDevice(parsed.dst_device, &dst_dev); in RecvFromRemoteAsync()
|
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
D | grpc_server_lib.cc | 97 delete worker_env_.device_mgr; in ~GrpcServer() 126 worker_env_.device_mgr = new DeviceMgr(worker_env_.local_devices); in Init()
|
D | rpc_rendezvous_mgr.cc | 230 s = sess->device_mgr->LookupDevice(parsed.dst_device, &dst_device); in RecvFromRemoteAsync()
|
/external/tensorflow/tensorflow/c/ |
D | c_api_internal.h | 134 const tensorflow::DeviceMgr* device_mgr; // Owned by session. member
|
/external/tensorflow/tensorflow/contrib/gdr/ |
D | gdr_rendezvous_mgr.cc | 158 Status s = sess->device_mgr->LookupDevice(parsed.dst_device, &dst_device); in RecvFromRemoteAsync()
|