| /external/tensorflow/tensorflow/core/common_runtime/ |
| rendezvous_mgr.h | 59 const DeviceMgr* device_mgr_; member in class:tensorflow::IntraProcessRendezvous
|
| process_function_library_runtime.h | 158 const DeviceMgr* const device_mgr_; member in class:tensorflow::ProcessFunctionLibraryRuntime
|
| process_function_library_runtime_test.cc | 40 device_mgr_.reset(new DeviceMgr(devices_));
46 device_mgr_.get(), Env::Default(), TF_GRAPH_DEF_VERSION, lib_def_.get(),
48 rendezvous_ = new IntraProcessRendezvous(device_mgr_.get());
107 std::unique_ptr<DeviceMgr> device_mgr_; member in class:tensorflow::__anon39502::ProcessFunctionLibraryRuntimeTest
|
| direct_session.h | 102 *output = device_mgr_.get();
273 const std::unique_ptr<const DeviceMgr> device_mgr_; member in class:tensorflow::DirectSession
|
| function.cc | 192 const DeviceMgr* const device_mgr_; member in class:tensorflow::FunctionLibraryRuntimeImpl
251 : device_mgr_(dmgr),
466 if (!device_mgr_->LookupDevice(options.target, &target_device).ok()) {
760 Rendezvous* rendezvous = new IntraProcessRendezvous(device_mgr_);
840 Rendezvous* rendezvous = new IntraProcessRendezvous(device_mgr_);
[all...] |
| function_test.cc | 149 device_mgr_.reset(new DeviceMgr(devices_));
151 device_mgr_.get(), Env::Default(), TF_GRAPH_DEF_VERSION, lib_def_.get(),
341 std::unique_ptr<DeviceMgr> device_mgr_; member in class:tensorflow::__anon39479::FunctionLibraryRuntimeTest
[all...] |
| /external/tensorflow/tensorflow/c/eager/ |
| runtime_test.cc | 41 device_mgr_.reset(new DeviceMgr({device}));
42 flib_runtime_ = NewFunctionLibraryRuntime(device_mgr_.get(), Env::Default(),
53 std::unique_ptr<DeviceMgr> device_mgr_; member in class:tensorflow::__anon38657::TestEnv
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| graph_mgr.h | 144 DeviceMgr* device_mgr_; member in class:tensorflow::GraphMgr
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| xla_compiler.h | 338 XlaCompilationDevice* device_; // Owned by device_mgr_
339 DeviceMgr device_mgr_; member in class:tensorflow::XlaCompiler
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| iterator_ops.cc | 91 : device_mgr_(std::move(device_mgr)),
217 // The following (device_mgr_, flib_def_, pflr_) are only used when the
220 std::unique_ptr<DeviceMgr> device_mgr_; member in class:tensorflow::__anon39744::IteratorResource
[all...] |
