Home
last modified time | relevance | path

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

123

/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Dstandalone.cc48 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()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Drendezvous_mgr.cc38 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()
H A Drendezvous_mgr.h57 inline void UpdateDeviceManager(DeviceMgr* device_mgr) { in UpdateDeviceManager()
H A Dcollective_test_util.h81 std::unique_ptr<tensorflow::DeviceMgr> device_mgr; member
H A Dfunction.cc298 const DeviceMgr* FunctionLibraryRuntimeOverlay::device_mgr() const { in device_mgr() function in tensorflow::FunctionLibraryRuntimeOverlay
374 const DeviceMgr* device_mgr() const override { return device_mgr_; } in device_mgr() function in tensorflow::FunctionLibraryRuntimeImpl
1410 const DeviceMgr* device_mgr, Env* env, const ConfigProto* config, in NewFunctionLibraryRuntime()
H A Dcollective_executor_mgr.cc98 const ConfigProto& config, const DeviceMgr* device_mgr, in CreateProdLocalCollectiveExecutorMgr()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/
H A Dtpu_global_init.cc56 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
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dcustom_device_test.cc98 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
H A Deager_operation_test.cc26 StaticDeviceMgr device_mgr(DeviceFactory::NewDevice( in TEST() local
51 StaticDeviceMgr device_mgr(DeviceFactory::NewDevice( in TEST() local
H A Dcontext_distributed_manager.cc107 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
H A Dcontext.cc101 int64_t step_id, DeviceMgr* device_mgr) { in FindOrCreate()
150 DeviceMgr* device_mgr, bool device_mgr_owned, Rendezvous* rendezvous, in EagerContext()
266 void EagerContext::ResetPFLR(const DeviceMgr* device_mgr, Env* env, in ResetPFLR()
1224 tensorflow::DeviceMgr* device_mgr) { in UpdateGlobalRendezvousDeviceManager()
1323 std::unique_ptr<ServerInterface> new_server, DeviceMgr* device_mgr, in StoreCollectiveOpsServer()
H A Dtensor_handle_test.cc37 StaticDeviceMgr device_mgr(DeviceFactory::NewDevice( in TEST() local
378 StaticDeviceMgr device_mgr(std::move(devices)); in TEST_F() local
H A Deager_op_rewrite_registry_test.cc51 StaticDeviceMgr device_mgr(DeviceFactory::NewDevice( in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/clusters/
H A Dsingle_machine.cc223 const DeviceMgr* device_mgr; in GetPeakMemoryUsage() local
372 const DeviceMgr* device_mgr; in ResetSession() local
452 const DeviceMgr* device_mgr; in ClearAllocatorStats() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/
H A Ddelegate_data_test.cc61 tensorflow::StaticDeviceMgr device_mgr(tensorflow::DeviceFactory::NewDevice( in TEST() local
206 tensorflow::StaticDeviceMgr device_mgr(tensorflow::DeviceFactory::NewDevice( in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Diterator_ops.cc92 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
H A Diterator_ops_test.cc48 std::unique_ptr<DeviceMgr> device_mgr; in GetIteratorResource() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/core_runtime/
H A Dop_handler_registry.cc33 const DeviceMgr* device_mgr) { in RegisterOpHandlers()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/backends/cpu/
H A Dcpu_registration.cc35 const DeviceMgr* device_mgr) { in RegisterCpuOpHandler()
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/eager/
H A Deager_service_impl.cc263 tensorflow::DeviceMgr* device_mgr = worker_session->device_mgr(); in CreateContext() local
407 const tensorflow::DeviceMgr* device_mgr = worker_session->device_mgr(); in UpdateContext() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/backends/gpu/
H A Dgpu_registration.cc35 const DeviceMgr* device_mgr) { in RegisterGpuOpHandler()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/fallback/
H A Dfallback_state.cc49 [](const int64, const DeviceMgr *device_mgr, Rendezvous **r) { in FallbackState()
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dworker_session.h40 DeviceMgr* device_mgr() { in device_mgr() function
H A Dworker_env.h56 DeviceMgr* device_mgr = nullptr; member
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
H A Dkernel_utils.cc28 DynamicDeviceMgr* device_mgr, const SessionOptions& session_opts, in InitEagerContext()

123