| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | dynamic_device_mgr_test.cc | 76 TF_CHECK_OK(dm->AddDevices(std::move(added_devices))); in TEST()
90 TF_CHECK_OK(dm->AddDevices(std::move(devices))); in TEST()
111 TF_CHECK_OK(dm->AddDevices(std::move(added_devices))); in TEST()
122 TF_CHECK_OK(dm->AddDevices(std::move(added_devices))); in TEST()
139 TF_CHECK_OK(dm->AddDevices(std::move(devices))); in TEST()
154 TF_CHECK_OK(dm->AddDevices(std::move(devices))); in TEST()
159 Status s = dm->AddDevices(std::move(added_devices)); in TEST()
173 TF_CHECK_OK(dm->AddDevices(std::move(devices))); in TEST()
190 TF_CHECK_OK(dm->AddDevices(std::move(devices))); in TEST()
207 TF_CHECK_OK(dm->AddDevices(std::move(devices))); in TEST()
[all …]
|
| D | dynamic_device_mgr.cc | 34 Status status = AddDevices(std::move(devices)); in DynamicDeviceMgr()
145 Status DynamicDeviceMgr::AddDevices( in AddDevices() function in tensorflow::DynamicDeviceMgr
|
| D | device_resolver_local_test.cc | 38 TF_CHECK_OK(DeviceFactory::AddDevices(options, task_name, &devices)); in DeviceResolverLocalTest()
|
| D | collective_executor_mgr_test.cc | 49 TF_CHECK_OK(DeviceFactory::AddDevices(options, task_name, &devices)); in CollectiveExecutorMgrTest()
|
| D | device_mgr.h | 141 Status AddDevices(std::vector<std::unique_ptr<Device>> devices);
|
| /external/tensorflow/tensorflow/tools/optimization/ |
| D | optimization_pass_runner.h | 42 return AddDevices(tensorflow::DEVICE_CPU, count); in AddCpus()
46 return AddDevices(tensorflow::DEVICE_GPU, count); in AddGpus()
50 Status AddDevices(absl::string_view type, int count);
|
| D | optimization_pass_runner.cc | 138 Status OptimizationPassRunner::AddDevices(absl::string_view type, int count) { in AddDevices() function in tensorflow::OptimizationPassRunner
|
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| D | kernel_utils.cc | 33 tensorflow::Status status = tensorflow::DeviceFactory::AddDevices( in InitEagerContext()
40 Status s = device_mgr->AddDevices(std::move(devices)); in InitEagerContext()
|
| D | kernel_utils.h | 133 llvm::Error AddDevices(std::vector<std::unique_ptr<Device>> devices) { in AddDevices() function
137 ->AddDevices(std::move(devices)); in AddDevices()
|
| /external/tensorflow/tensorflow/python/client/ |
| D | device_lib_wrapper.cc | 41 tensorflow::MaybeRaiseFromStatus(tensorflow::DeviceFactory::AddDevices( in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | mark_for_compilation_pass_test_helper.cc | 47 TF_RETURN_IF_ERROR(DeviceFactory::AddDevices(session_options, "", &devices)); in MarkForCompilation()
|
| D | xla_kernel_creator_test.cc | 66 TF_CHECK_OK(DeviceFactory::AddDevices( in Init()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | grappler_item.h | 113 Status AddDevices(const GrapplerItem& other);
|
| /external/tensorflow/tensorflow/core/tfrt/fallback/ |
| D | fallback_state.cc | 30 TF_RETURN_IF_ERROR(DeviceFactory::AddDevices( in Create()
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | tf_optimizer_wrapper.cc | 45 if (!tensorflow::DeviceFactory::AddDevices(options, "", &devices).ok()) { in DetectDevices()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_factory.h | 54 static Status AddDevices(const SessionOptions& options,
|
| /external/tensorflow/tensorflow/core/function/ |
| D | runtime_client.cc | 60 Status&& device_init_status = DeviceFactory::AddDevices( in GlobalEagerContext()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | session_mgr.cc | 163 remote_devices->AddDevices(std::move(cluster_devices))); in CreateSession()
179 remote_devices->AddDevices(std::move(cluster_devices))); in CreateSession()
|
| D | rpc_collective_executor_mgr_test.cc | 49 TF_CHECK_OK(DeviceFactory::AddDevices(options, task_name, &devices)); in RpcCollectiveExecutorMgrTest()
|
| D | worker_session.cc | 155 remote_device_mgr_->AddDevices(std::move(added_remote_devices))); in UpdateWorkerCacheAndDevices()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | immediate_execution_context.h | 112 virtual Status AddDevices(std::vector<std::unique_ptr<Device>> devices) = 0;
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute_node_test.cc | 63 TF_ASSERT_OK(remote_device_mgr->AddDevices(std::move(remote_devices))); in TEST()
|
| D | placement_test.cc | 105 TF_CHECK_OK(device_manager_->AddDevices(std::move(added_devices))); in InitDeviceManager()
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
| D | saved_variable_loading_test.cc | 47 TF_CHECK_OK(DeviceFactory::AddDevices( in SavedVariableLoadingTest()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | shape_optimizer_test.cc | 84 ->AddDevices(session_options, "", &devices)); in TEST_F()
|