| /external/tensorflow/tensorflow/core/common_runtime/data/ |
| D | standalone.cc | 50 std::unique_ptr<DeviceMgr> device_mgr = MakeUnique<DeviceMgr>( in FromGraph() local
111 Dataset::Dataset(DatasetBase* dataset, DeviceMgr* device_mgr, in Dataset()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | delegate_data.cc | 38 std::unique_ptr<tensorflow::DeviceMgr> device_mgr = in Prepare() local
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine.cc | 224 const DeviceMgr* device_mgr; in GetPeakMemoryUsage() local
373 const DeviceMgr* device_mgr; in ResetSession() local
453 const DeviceMgr* device_mgr; in ClearAllocatorStats() local
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | worker_session.h | 48 DeviceMgr* device_mgr() { in device_mgr() function
|
| D | worker_env.h | 56 DeviceMgr* device_mgr = nullptr; member
|
| D | session_mgr.cc | 87 auto device_mgr = MakeUnique<DeviceMgr>(std::move(renamed_devices)); in CreateSession() local
|
| D | worker_session.cc | 98 std::unique_ptr<DeviceMgr> device_mgr, in WorkerSession()
|
| D | graph_mgr.cc | 56 GraphMgr::GraphMgr(const WorkerEnv* worker_env, DeviceMgr* device_mgr) in GraphMgr()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.cc | 49 std::unique_ptr<const DeviceMgr> device_mgr, in EagerContext()
56 bool async, const DeviceMgr* device_mgr, in EagerContext()
430 std::unique_ptr<ServerInterface> server, DeviceMgr* device_mgr, in StoreCollectiveOpsServer()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | device.h | 179 DeviceMgr* device_mgr() const { return device_mgr_; } in device_mgr() function
|
| D | rendezvous_mgr.cc | 34 IntraProcessRendezvous::IntraProcessRendezvous(const DeviceMgr* device_mgr) in IntraProcessRendezvous()
|
| D | process_function_library_runtime.h | 140 const DeviceMgr* device_mgr() { return device_mgr_; } in device_mgr() function
|
| D | function.cc | 277 const DeviceMgr* FunctionLibraryRuntimeOverlay::device_mgr() const { in device_mgr() function in tensorflow::FunctionLibraryRuntimeOverlay
349 const DeviceMgr* device_mgr() const override { return device_mgr_; } in device_mgr() function in tensorflow::FunctionLibraryRuntimeImpl
1220 const DeviceMgr* device_mgr, Env* env, Device* device, in NewFunctionLibraryRuntime()
1231 const DeviceMgr* device_mgr, Env* env, Device* device, in NewFunctionLibraryRuntime()
|
| D | process_function_library_runtime.cc | 59 const DeviceMgr* device_mgr, Env* env, int graph_def_version, in ProcessFunctionLibraryRuntime()
84 const DeviceMgr* device_mgr, Env* env, int graph_def_version, in ProcessFunctionLibraryRuntime()
|
| D | direct_session.cc | 275 const DeviceMgr* device_mgr, in DirectSession()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.cc | 59 std::unique_ptr<DeviceMgr> device_mgr, in IteratorResource()
422 std::unique_ptr<DeviceMgr> device_mgr(nullptr); in Compute() local
477 OpKernelContext* ctx, std::unique_ptr<DeviceMgr>* device_mgr, in CreatePrivateFLR()
510 std::unique_ptr<DeviceMgr> device_mgr(nullptr); in Compute() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph_test.cc | 192 const DeviceMgr* device_mgr = nullptr; in TEST() local
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 83 std::unique_ptr<tensorflow::DeviceMgr>* device_mgr) { in GetAllRemoteDevices()
227 auto* device_mgr = grpc_server->worker_env()->device_mgr; in UpdateTFE_ContextWithServerDef() local
342 std::unique_ptr<tensorflow::DeviceMgr> device_mgr( in TFE_NewContext() local
355 const tensorflow::DeviceMgr* device_mgr = nullptr; in TFE_NewContextFromSession() local
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.cc | 105 std::unique_ptr<tensorflow::DeviceMgr> device_mgr = in CreateContext() local
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_subgraphs_pass.cc | 2560 std::unique_ptr<DeviceMgr> device_mgr = in Run() local
|
| D | encapsulate_subgraphs_pass_test.cc | 510 auto device_mgr = absl::make_unique<DeviceMgr>(std::move(devices)); in Encapsulate() local
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | import_tensorflow.cc | 2156 tensorflow::DeviceMgr device_mgr(std::move(devices)); in InlineAllFunctions() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 3459 tensorflow::DeviceMgr device_mgr(std::move(devices)); in main() local
|