Home
last modified time | relevance | path

Searched defs:device_mgr (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/
Dworker_session.h46 const std::unique_ptr<DeviceMgr> device_mgr; member
Dworker_env.h55 DeviceMgr* device_mgr = nullptr; member
Dworker_session.cc93 std::unique_ptr<DeviceMgr> device_mgr, in WorkerSession()
Dsession_mgr.cc79 std::unique_ptr<DeviceMgr> device_mgr(new DeviceMgr(renamed_devices)); in CreateSession() local
Dgraph_mgr.cc52 GraphMgr::GraphMgr(const WorkerEnv* worker_env, DeviceMgr* device_mgr) in GraphMgr()
/external/tensorflow/tensorflow/core/grappler/clusters/
Dsingle_machine.cc211 const DeviceMgr* device_mgr; in GetPeakMemoryUsage() local
400 const DeviceMgr* device_mgr; in ClearAllocatorStats() local
/external/tensorflow/tensorflow/core/common_runtime/
Drendezvous_mgr.cc34 IntraProcessRendezvous::IntraProcessRendezvous(const DeviceMgr* device_mgr) in IntraProcessRendezvous()
Dprocess_function_library_runtime.cc29 const DeviceMgr* device_mgr, Env* env, int graph_def_version, in ProcessFunctionLibraryRuntime()
51 const DeviceMgr* device_mgr, Env* env, int graph_def_version, in ProcessFunctionLibraryRuntime()
Dfunction.cc945 const DeviceMgr* device_mgr, Env* env, Device* device, in NewFunctionLibraryRuntime()
956 const DeviceMgr* device_mgr, Env* env, Device* device, in NewFunctionLibraryRuntime()
Ddirect_session.cc254 const DeviceMgr* device_mgr, in DirectSession()
/external/tensorflow/tensorflow/c/
Dc_api_internal.h134 const tensorflow::DeviceMgr* device_mgr; // Owned by session. member
/external/tensorflow/tensorflow/core/kernels/data/
Diterator_ops.cc87 std::unique_ptr<DeviceMgr> device_mgr, in IteratorResource()
467 std::unique_ptr<DeviceMgr> device_mgr(nullptr); in Compute() local
542 OpKernelContext* ctx, std::unique_ptr<DeviceMgr>* device_mgr, in CreatePrivateFLR()
/external/tensorflow/tensorflow/contrib/lite/toco/
Dimport_tensorflow.cc1788 tensorflow::DeviceMgr device_mgr(devices); in InlineAllFunctions() local
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc3292 tensorflow::DeviceMgr device_mgr(devices); in main() local