Home
last modified time | relevance | path

Searched refs:AddDevices (Results 1 – 25 of 33) sorted by relevance

12

/external/tensorflow/tensorflow/lite/delegates/flex/
Ddelegate_data.cc35 TF_RETURN_IF_ERROR(tensorflow::DeviceFactory::AddDevices( in Prepare()
/external/tensorflow/tensorflow/compiler/jit/
Dmark_for_compilation_pass_test_helper.cc45 TF_RETURN_IF_ERROR(DeviceFactory::AddDevices(session_options, "", &devices)); in MarkForCompilation()
Dcreate_xla_launch_op_test.cc63 TF_CHECK_OK(DeviceFactory::AddDevices( in Init()
Dbuild_xla_ops_pass_test.cc39 CHECK(DeviceFactory::AddDevices( in SetUp()
Dpartially_decluster_pass_test.cc397 TF_ASSERT_OK(DeviceFactory::AddDevices( in TEST()
/external/tensorflow/tensorflow/core/common_runtime/
Ddevice_factory.h41 static Status AddDevices(const SessionOptions& options,
Ddevice_resolver_local_test.cc39 TF_CHECK_OK(DeviceFactory::AddDevices(options, task_name, &devices)); in DeviceResolverLocalTest()
Dcollective_executor_mgr_test.cc42 TF_CHECK_OK(DeviceFactory::AddDevices(options, task_name, &devices)); in CollectiveExecutorMgrTest()
Ddevice_factory.cc93 Status DeviceFactory::AddDevices( in AddDevices() function in tensorflow::DeviceFactory
Dcollective_rma_local_test.cc46 TF_CHECK_OK(DeviceFactory::AddDevices(options, kTaskName, &devices)); in CollectiveRemoteAccessLocalTest()
Dpartitioning_utils_test.cc46 TF_CHECK_OK(DeviceFactory::AddDevices(options, "/job:a/replica:0/task:0", in SetUp()
Dfunction_threadpool_test.cc58 TF_CHECK_OK(DeviceFactory::AddDevices( in Init()
Dcollective_param_resolver_local_test.cc41 TF_CHECK_OK(DeviceFactory::AddDevices(options, task_name, &devices)); in CollectiveParamResolverLocalTest()
Dprocess_function_library_runtime_test.cc73 TF_CHECK_OK(DeviceFactory::AddDevices(options, "/job:a/replica:0/task:0", in Init()
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item.h109 Status AddDevices(const GrapplerItem& other);
Dgrappler_item.cc152 Status GrapplerItem::AddDevices(const GrapplerItem& other) { in AddDevices() function in tensorflow::grappler::GrapplerItem
/external/tensorflow/tensorflow/python/client/
Ddevice_lib.i52 Status status = DeviceFactory::AddDevices( in ListDevicesWithSessionConfig()
/external/tensorflow/tensorflow/python/grappler/
Dtf_optimizer.i78 tensorflow::Status status = tensorflow::DeviceFactory::AddDevices(options, "", &devices); in DetectDevices()
/external/tensorflow/tensorflow/core/distributed_runtime/
Drpc_collective_executor_mgr_test.cc46 TF_CHECK_OK(DeviceFactory::AddDevices(options, task_name, &devices)); in RpcCollectiveExecutorMgrTest()
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Deager_service_impl.cc93 TF_RETURN_IF_ERROR(tensorflow::DeviceFactory::AddDevices( in CreateContext()
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_test_base.cc148 TF_RETURN_IF_ERROR(DeviceFactory::AddDevices( in InitFunctionLibraryRuntime()
/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_reducer_test.cc85 ->AddDevices(session_options, "", &all_devices); in InitGPUDevices()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_server_lib.cc159 DeviceFactory::AddDevices(sess_opts, name_prefix, &devices)); in Init()
/external/tensorflow/tensorflow/core/nccl/
Dnccl_manager_test.cc38 ->AddDevices(session_options, "", &devices); in GetGPUDevices()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmeta_optimizer.cc611 Status added_devices = func_item.AddDevices(item); in Optimize()

12