| /external/tensorflow/tensorflow/core/data/ |
| D | standalone.cc | 48 ProcessFunctionLibraryRuntime* pflr, DeviceMgr* device_mgr, in CreateParams()
72 auto device_mgr = absl::make_unique<StaticDeviceMgr>(DeviceFactory::NewDevice( in FromGraph() local
85 [](const int64_t, const DeviceMgr* device_mgr, Rendezvous** r) { in FromGraph()
167 Dataset::Dataset(DatasetBase* dataset, DeviceMgr* device_mgr, in Dataset()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | rendezvous_mgr.cc | 37 void SameWorkerRecvDone(const DeviceMgr* device_mgr, in SameWorkerRecvDone()
118 void IntraProcessRecvAsyncImpl(const DeviceMgr* device_mgr, in IntraProcessRecvAsyncImpl()
156 const DeviceMgr* device_mgr) in RefCountedIntraProcessRendezvous()
181 const DeviceMgr* device_mgr) in PrivateIntraProcessRendezvous()
|
| 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/core/common_runtime/eager/ |
| 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_operation_test.cc | 26 StaticDeviceMgr device_mgr(DeviceFactory::NewDevice( in TEST() local
|
| D | eager_op_rewrite_registry_test.cc | 50 StaticDeviceMgr device_mgr(DeviceFactory::NewDevice( in TEST() local
|
| D | execute_node_test.cc | 51 StaticDeviceMgr device_mgr( in TEST() 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
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | delegate_data_test.cc | 60 tensorflow::StaticDeviceMgr device_mgr(tensorflow::DeviceFactory::NewDevice( in TEST() local
204 tensorflow::StaticDeviceMgr device_mgr(tensorflow::DeviceFactory::NewDevice( in TEST() local
|
| D | delegate_data.cc | 208 auto device_mgr = in Prepare() local
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine.cc | 222 const DeviceMgr* device_mgr; in GetPeakMemoryUsage() local
371 const DeviceMgr* device_mgr; in ResetSession() local
451 const DeviceMgr* device_mgr; in ClearAllocatorStats() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | upgrade_graph.cc | 115 static const auto* const device_mgr = []() -> const DeviceMgr* { in GetStaticDeviceMgr() local
164 const auto* device_mgr = GetStaticDeviceMgr(); in RunGrappler() 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/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/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/kernels/data/ |
| D | iterator_ops.cc | 87 std::unique_ptr<DeviceMgr> device_mgr, in IteratorResource()
489 std::unique_ptr<DeviceMgr> device_mgr(nullptr); in Compute() local
544 OpKernelContext* ctx, std::unique_ptr<DeviceMgr>* device_mgr, in CreatePrivateFLR()
586 std::unique_ptr<DeviceMgr> device_mgr(nullptr); in CreateResource() local
|
| /external/tensorflow/tensorflow/core/tfrt/eager/ |
| D | transform_graph_function.cc | 43 const DeviceMgr* device_mgr = eager_ctx->local_device_mgr(); in TransformGraphFunction() local
|
| /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
|