| /external/tensorflow/tensorflow/core/data/ |
| D | standalone.cc | 48 ProcessFunctionLibraryRuntime* pflr, DeviceMgr* device_mgr, in CreateParams()
72 auto device_mgr = std::make_unique<StaticDeviceMgr>(DeviceFactory::NewDevice( in FromGraph() local
85 [](const int64_t, const DeviceMgr* device_mgr, Rendezvous** r) { in FromGraph()
168 DeviceMgr* device_mgr, ProcessFunctionLibraryRuntime* pflr, in Dataset()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | eager_operation_test.cc | 26 StaticDeviceMgr device_mgr(DeviceFactory::NewDevice( in TEST() local
51 StaticDeviceMgr device_mgr(DeviceFactory::NewDevice( in TEST() local
|
| D | custom_device_test.cc | 98 StaticDeviceMgr device_mgr(DeviceFactory::NewDevice( in TEST() local
127 StaticDeviceMgr device_mgr(DeviceFactory::NewDevice( in TEST() local
145 StaticDeviceMgr device_mgr(DeviceFactory::NewDevice( in TEST() local
|
| D | eager_op_rewrite_registry_test.cc | 51 StaticDeviceMgr device_mgr(DeviceFactory::NewDevice( in TEST() local
|
| D | execute_node_test.cc | 56 StaticDeviceMgr device_mgr( in TEST() local
|
| D | mkl_eager_op_rewrite_test.cc | 37 std::unique_ptr<DeviceMgr> device_mgr = in CreateOp() local
|
| D | context_distributed_manager.cc | 107 std::unique_ptr<DynamicDeviceMgr>* device_mgr) { in GetAllRemoteDevices()
423 DeviceMgr* device_mgr = AreLocalDevicesCompatible(context, server_def) in UpdateContextWithServerDef() local
604 auto* device_mgr = server->worker_env()->device_mgr; in UpdateContextWithServerDef() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | rendezvous_mgr.cc | 38 void SameWorkerRecvDone(const DeviceMgr* device_mgr, in SameWorkerRecvDone()
132 void IntraProcessRecvAsyncImpl(const DeviceMgr* device_mgr, in IntraProcessRecvAsyncImpl()
170 const DeviceMgr* device_mgr) in RefCountedIntraProcessRendezvous()
199 const DeviceMgr* device_mgr) in PrivateIntraProcessRendezvous()
|
| D | rendezvous_mgr.h | 57 inline void UpdateDeviceManager(DeviceMgr* device_mgr) { in UpdateDeviceManager()
|
| D | collective_test_util.h | 81 std::unique_ptr<tensorflow::DeviceMgr> device_mgr; member
|
| D | collective_executor_mgr.cc | 98 const ConfigProto& config, const DeviceMgr* device_mgr, in CreateProdLocalCollectiveExecutorMgr()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_global_init.cc | 56 Status CreateDeviceMgr(Env* env, std::unique_ptr<DeviceMgr>* device_mgr) { in CreateDeviceMgr()
70 void DeviceSetFromDeviceMgr(const DeviceMgr& device_mgr, in DeviceSetFromDeviceMgr()
203 std::unique_ptr<DeviceMgr> device_mgr; in InitializeTPUSystemGlobally() local
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | delegate_data_test.cc | 61 tensorflow::StaticDeviceMgr device_mgr(tensorflow::DeviceFactory::NewDevice( in TEST() local
206 tensorflow::StaticDeviceMgr device_mgr(tensorflow::DeviceFactory::NewDevice( in TEST() local
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine.cc | 223 const DeviceMgr* device_mgr; in GetPeakMemoryUsage() local
372 const DeviceMgr* device_mgr; in ResetSession() local
452 const DeviceMgr* device_mgr; in ClearAllocatorStats() local
|
| /external/tensorflow/tensorflow/core/tfrt/eager/core_runtime/ |
| D | op_handler_registry.cc | 33 const DeviceMgr* device_mgr) { in RegisterOpHandlers()
|
| /external/tensorflow/tensorflow/core/tfrt/eager/backends/cpu/ |
| D | cpu_registration.cc | 35 const DeviceMgr* device_mgr) { in RegisterCpuOpHandler()
|
| /external/tensorflow/tensorflow/core/tfrt/eager/backends/gpu/ |
| D | gpu_registration.cc | 35 const DeviceMgr* device_mgr) { in RegisterGpuOpHandler()
|
| /external/tensorflow/tensorflow/core/tfrt/fallback/ |
| D | fallback_state.cc | 49 [](const int64, const DeviceMgr *device_mgr, Rendezvous **r) { in FallbackState()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | worker_session.h | 40 DeviceMgr* device_mgr() { in device_mgr() function
|
| D | worker_env.h | 56 DeviceMgr* device_mgr = nullptr; member
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops_test.cc | 48 std::unique_ptr<DeviceMgr> device_mgr; in GetIteratorResource() local
|
| D | iterator_ops.cc | 92 std::unique_ptr<DeviceMgr> device_mgr, in IteratorResource()
467 std::unique_ptr<DeviceMgr> device_mgr(nullptr); in Compute() local
522 OpKernelContext* ctx, std::unique_ptr<DeviceMgr>* device_mgr, in CreatePrivateFLR()
574 std::unique_ptr<DeviceMgr> device_mgr(nullptr); in CreateResource() local
|
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| D | kernel_utils.cc | 28 DynamicDeviceMgr* device_mgr, const SessionOptions& session_opts, in InitEagerContext()
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
| D | test_utils.cc | 44 EagerContextPtr CreateTestingEagerContext(DeviceMgr* device_mgr) { in CreateTestingEagerContext()
|
| /external/tensorflow/tensorflow/core/tfrt/eager/ |
| D | transform_graph_function.cc | 43 const DeviceMgr* device_mgr = eager_ctx->local_device_mgr(); in TransformGraphFunction() local
|