Home
last modified time | relevance | path

Searched refs:AddDevice (Results 1 – 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/tools/optimization/
Dgpu_optimization_pass_runner_main.cc36 TF_CHECK_OK(runner.AddDevice( in main()
39 TF_CHECK_OK(runner.AddDevice( in main()
42 TF_CHECK_OK(runner.AddDevice( in main()
45 TF_CHECK_OK(runner.AddDevice( in main()
48 TF_CHECK_OK(runner.AddDevice( in main()
51 TF_CHECK_OK(runner.AddDevice( in main()
Doptimization_pass_runner.cc127 device_set.AddDevice(device.get()); in RunMain()
161 Status OptimizationPassRunner::AddDevice(const string& name, in AddDevice() function in tensorflow::OptimizationPassRunner
Doptimization_pass_runner.h39 Status AddDevice(const string& name, const string& type);
/external/tensorflow/tensorflow/core/common_runtime/
Ddevice_set_test.cc42 void AddDevice(const char* type, const char* name) { in AddDevice() function in tensorflow::__anonf0ab42df0111::DeviceSetTest
45 devices_.AddDevice(d); in AddDevice()
77 AddDevice("d1", "/job:a/replica:0/task:0/device:d1:0"); in TEST_F()
80 AddDevice("d1", "/job:a/replica:0/task:0/device:d1:1"); in TEST_F()
84 AddDevice("d2", "/job:a/replica:0/task:0/device:d2:0"); in TEST_F()
89 AddDevice("d3", "/job:a/replica:0/task:0/device:d3:0"); in TEST_F()
Ddevice_set.h38 void AddDevice(Device* device);
Ddevice_set.cc33 void DeviceSet::AddDevice(Device* device) { in AddDevice() function in tensorflow::DeviceSet
Dpartitioning_utils_test.cc53 device_set_.AddDevice(d); in SetUp()
Dplacer_test.cc209 devices_.AddDevice(local_devices_.back().get()); in PlacerTest()
213 devices_.AddDevice(local_devices_.back().get()); in PlacerTest()
1264 heterogeneous.AddDevice(gpu.get()); in TEST_F()
1267 heterogeneous.AddDevice(cpu.get()); in TEST_F()
1365 cpu_only.AddDevice(cpu.get()); in TEST_F()
Dprocess_function_library_runtime.cc570 device_set.AddDevice(d); in InstantiateMultiDevice()
/external/tensorflow/tensorflow/core/common_runtime/sycl/
Dsycl_device.h42 AddDevice(device); in GSYCLInterface()
56 AddDevice(device); in GSYCLInterface()
70 AddDevice(device); in GSYCLInterface()
115 void AddDevice(const cl::sycl::device& d) { in AddDevice() function
/external/perfetto/src/traced/probes/ftrace/
Dftrace_controller_unittest.cc570 TEST(FtraceMetadataTest, AddDevice) { in TEST() argument
572 metadata.AddDevice(1); in TEST()
574 metadata.AddDevice(3); in TEST()
581 metadata.AddDevice(3); in TEST()
585 metadata.AddDevice(4); in TEST()
590 metadata.AddDevice(5); in TEST()
Dftrace_metadata.h49 void AddDevice(BlockDeviceID);
Dftrace_metadata.cc31 void FtraceMetadata::AddDevice(BlockDeviceID device_id) { in AddDevice() function in perfetto::FtraceMetadata
Dcpu_reader.h114 metadata->AddDevice(dev_id); in ReadDevId()
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item.cc136 Status GrapplerItem::AddDevice(const string& device) { in AddDevice() function in tensorflow::grappler::GrapplerItem
155 Status added = AddDevice(device); in AddDevices()
168 Status added = AddDevice(node.device()); in InferDevicesFromGraph()
Dgrappler_item.h107 Status AddDevice(const string& device);
/external/webrtc/talk/media/webrtc/
Dfakewebrtcdeviceinfo.h47 void AddDevice(const std::string& device_name, const std::string& device_id) { in AddDevice() function
Dwebrtcvideocapturer_unittest.cc52 factory_->device_info.AddDevice(kTestDeviceName, kTestDeviceId); in WebRtcVideoCapturerTest()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmaster.cc431 device_set->AddDevice(d.get()); in CreateSession()
449 device_set->AddDevice(d.get()); in CreateSession()
453 device_set->AddDevice(d); in CreateSession()
Dtest_utils.h131 void AddDevice(const string& device_name, const DeviceLocality& dev_loc) { in AddDevice() function
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph_test.cc195 device_set.AddDevice(d); in TEST()
/external/tensorflow/tensorflow/core/grappler/clusters/
Dsingle_machine.cc376 device_set_->AddDevice(d); in ResetSession()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcontext.cc113 ds.AddDevice(d); in InitDeviceMapAndAsync()
/external/tensorflow/tensorflow/compiler/jit/
Dincrease_dynamism_for_auto_jit_pass_test.cc70 device_set->AddDevice(device.get()); in IncreaseDynamismForAutoJit()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmeta_optimizer.cc731 Status added_device = item.AddDevice(d->name()); in OptimizeGraph()

12