| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | rendezvous_mgr.cc | 36 void SameWorkerRecvDone(const DeviceMgr* device_mgr, in SameWorkerRecvDone()
115 void IntraProcessRecvAsyncImpl(const DeviceMgr* device_mgr, in IntraProcessRecvAsyncImpl()
153 const DeviceMgr* device_mgr) in RefCountedIntraProcessRendezvous()
178 const DeviceMgr* device_mgr) in PrivateIntraProcessRendezvous()
|
| D | process_function_library_runtime.h | 203 const DeviceMgr* device_mgr() { return device_mgr_; } in device_mgr() function
|
| D | function_threadpool_test.cc | 70 [](const int64, const DeviceMgr* device_mgr, Rendezvous** r) { in Init()
|
| D | function.cc | 294 const DeviceMgr* FunctionLibraryRuntimeOverlay::device_mgr() const { in device_mgr() function in tensorflow::FunctionLibraryRuntimeOverlay
370 const DeviceMgr* device_mgr() const override { return device_mgr_; } in device_mgr() function in tensorflow::FunctionLibraryRuntimeImpl
1373 const DeviceMgr* device_mgr, Env* env, const ConfigProto* config, in NewFunctionLibraryRuntime()
|
| /external/tensorflow/tensorflow/core/data/ |
| D | standalone.cc | 51 auto device_mgr = absl::make_unique<StaticDeviceMgr>(DeviceFactory::NewDevice( in FromGraph() local
64 [](const int64, const DeviceMgr* device_mgr, Rendezvous** r) { in FromGraph()
144 Dataset::Dataset(DatasetBase* dataset, DeviceMgr* device_mgr, in Dataset()
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine.cc | 221 const DeviceMgr* device_mgr; in GetPeakMemoryUsage() local
370 const DeviceMgr* device_mgr; in ResetSession() local
450 const DeviceMgr* device_mgr; in ClearAllocatorStats() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | upgrade_graph.cc | 113 static const auto* const device_mgr = []() -> const DeviceMgr* { in GetStaticDeviceMgr() local
162 const auto* device_mgr = GetStaticDeviceMgr(); in RunGrappler() local
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | eager_operation_test.cc | 26 StaticDeviceMgr device_mgr(DeviceFactory::NewDevice( in TEST() local
|
| D | custom_device_test.cc | 87 StaticDeviceMgr device_mgr(DeviceFactory::NewDevice( in TEST() local
112 StaticDeviceMgr device_mgr(DeviceFactory::NewDevice( in TEST() local
|
| D | eager_op_rewrite_registry_test.cc | 46 StaticDeviceMgr device_mgr(DeviceFactory::NewDevice( in TEST() local
|
| D | execute_node_test.cc | 51 StaticDeviceMgr device_mgr( in TEST() local
|
| D | context_distributed_manager.cc | 99 std::unique_ptr<tensorflow::DynamicDeviceMgr>* device_mgr) { in GetAllRemoteDevices()
426 const tensorflow::DeviceMgr* device_mgr = in UpdateContextWithServerDef() local
608 auto* device_mgr = grpc_server->worker_env()->device_mgr; in UpdateContextWithServerDef() local
|
| D | mkl_eager_op_rewrite_test.cc | 37 std::unique_ptr<DeviceMgr> device_mgr = in CreateOp() local
|
| D | tensor_handle_test.cc | 37 StaticDeviceMgr device_mgr(DeviceFactory::NewDevice( in TEST() local
378 StaticDeviceMgr device_mgr(std::move(devices)); in TEST_F() local
|
| D | context.cc | 79 const DeviceMgr* device_mgr, bool device_mgr_owned, Rendezvous* rendezvous, in EagerContext()
185 void EagerContext::ResetPFLR(const DeviceMgr* device_mgr, Env* env, in ResetPFLR()
1066 std::unique_ptr<ServerInterface> new_server, const DeviceMgr* device_mgr, in StoreCollectiveOpsServer()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | delegate_data.cc | 41 auto device_mgr = in Prepare() local
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | worker_session.h | 40 const DeviceMgr* device_mgr() { in device_mgr() function
|
| D | worker_env.h | 56 const DeviceMgr* device_mgr = nullptr; member
|
| D | worker_session.cc | 110 std::unique_ptr<DeviceMgr> device_mgr, std::unique_ptr<GraphMgr> graph_mgr, in WorkerSession()
|
| D | session_mgr.cc | 139 auto device_mgr = MakeUnique<StaticDeviceMgr>(std::move(renamed_devices)); in CreateSession() local
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
| D | test_utils.cc | 44 EagerContextPtr CreateTestingEagerContext(DeviceMgr* device_mgr) { in CreateTestingEagerContext()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_mgr_test.cc | 51 auto device_mgr = absl::make_unique<StaticDeviceMgr>(std::move(devices)); in RemoteMgrTest() local
|
| D | eager_service_impl.cc | 258 const tensorflow::DeviceMgr* device_mgr = worker_session->device_mgr(); in CreateContext() local
373 const tensorflow::DeviceMgr* device_mgr = worker_session->device_mgr(); in UpdateContext() local
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.cc | 82 std::unique_ptr<DeviceMgr> device_mgr, in IteratorResource()
475 std::unique_ptr<DeviceMgr> device_mgr(nullptr); in Compute() local
530 OpKernelContext* ctx, std::unique_ptr<DeviceMgr>* device_mgr, in CreatePrivateFLR()
572 std::unique_ptr<DeviceMgr> device_mgr(nullptr); in CreateResource() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph_test.cc | 119 const DeviceMgr* device_mgr = nullptr; in TEST() local
|