Home
last modified time | relevance | path

Searched refs:FakeDevice (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/tools/optimization/
Doptimization_pass_runner.cc47 class FakeDevice : public Device { class
49 explicit FakeDevice(const DeviceAttributes& device_attributes) in FakeDevice() function in tensorflow::__anone07b0ebc0111::FakeDevice
57 Status FakeDevice::Sync() { in Sync()
61 std::unique_ptr<Device> FakeDevice::Make(const string& name, in Make()
66 return std::unique_ptr<Device>(new FakeDevice(device_attributes)); in Make()
140 devices_.push_back(FakeDevice::Make( in AddDevices()
143 devices_.push_back(FakeDevice::Make( in AddDevices()
/external/tensorflow/tensorflow/core/common_runtime/
Ddevice_mgr_test.cc31 class FakeDevice : public Device { in CreateDevice() class
33 explicit FakeDevice(const DeviceAttributes& attr) : Device(nullptr, attr) {} in CreateDevice() function in tensorflow::__anon7c4d83f70111::CreateDevice::FakeDevice
40 return new FakeDevice(attr); in CreateDevice()
Ddynamic_device_mgr_test.cc35 class FakeDevice : public Device { in CreateDevice() class
37 explicit FakeDevice(const DeviceAttributes& attr) : Device(nullptr, attr) {} in CreateDevice() function in tensorflow::__anonc282fa3b0111::CreateDevice::FakeDevice
42 class FakeDeviceWithDestructorNotification : public FakeDevice { in CreateDevice()
46 : FakeDevice(attr), n_(n) {} in CreateDevice()
63 return new FakeDevice(attr); in CreateDevice()
Ddevice_set_test.cc29 class FakeDevice : public Device { in Dev() class
31 explicit FakeDevice(const DeviceAttributes& attr) : Device(nullptr, attr) {} in Dev() function in tensorflow::__anon7d5d833d0111::Dev::FakeDevice
38 return new FakeDevice(attr); in Dev()
Dbuf_rendezvous_test.cc34 class FakeDevice : public Device { in NewDevice() class
36 explicit FakeDevice(const DeviceAttributes& attrs) in NewDevice() function in tensorflow::__anon5d7062b30111::BufRendezvousTest::NewDevice::FakeDevice
45 return absl::make_unique<FakeDevice>(attrs); in NewDevice()
Dplacer_test.cc85 class FakeDevice : public Device { class
87 explicit FakeDevice(const DeviceAttributes& device_attributes) in FakeDevice() function in tensorflow::__anon97448ef90111::FakeDevice
100 return std::unique_ptr<Device>(new FakeDevice(device_attributes)); in MakeDevice()
225 local_devices_.emplace_back(FakeDevice::MakeCPU( in PlacerTest()
229 local_devices_.emplace_back(FakeDevice::MakeGPU(strings::StrCat( in PlacerTest()
233 local_devices_.emplace_back(FakeDevice::MakeDevice( in PlacerTest()
236 local_devices_.emplace_back(FakeDevice::MakeDevice( in PlacerTest()
1472 FakeDevice::MakeGPU("/job:b/replica:0/task:0/device:FakeGPU:0")); in TEST_F()
1475 FakeDevice::MakeCPU("/job:b/replica:0/task:1/device:FakeCPU:0")); in TEST_F()
1574 std::unique_ptr<Device> gpu(FakeDevice::MakeGPU(gpu_device)); in TEST_F()
[all …]
Dconstant_folding_test.cc96 class FakeDevice : public Device { class
98 explicit FakeDevice(const DeviceAttributes& device_attributes) in FakeDevice() function in tensorflow::__anon6fd1b14e0111::FakeDevice
110 return std::unique_ptr<Device>(new FakeDevice(device_attributes)); in Make()
652 FakeDevice::Make("/job:tpu_worker/replica:0/task:0/device:GPU:0", in TEST_F()
/external/skqp/tests/
DLayerDrawLooperTest.cpp28 class FakeDevice : public SkBitmapDevice { class
30 FakeDevice() : INHERITED(make_bm(100, 100), SkSurfaceProps(0, kUnknown_SkPixelGeometry), in FakeDevice() function in FakeDevice
58 FakeDevice device; in test_frontToBack()
97 FakeDevice device; in test_backToFront()
136 FakeDevice device; in test_mixed()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Ddevice_util_test.cc41 class FakeDevice : public Device { class
43 explicit FakeDevice(const DeviceAttributes& device_attributes) in FakeDevice() function in tensorflow::__anon39ebad2b0111::FakeDevice
57 return std::make_unique<FakeDevice>(device_attributes); in Make()
71 devices.push_back(FakeDevice::Make(cpu0)); in TEST()
72 devices.push_back(FakeDevice::Make(gpu0, "compute capability: 7.0")); in TEST()
73 devices.push_back(FakeDevice::Make(gpu1)); in TEST()
/external/tensorflow/tensorflow/core/distributed_runtime/
Ddevice_resolver_distributed_test.cc38 class FakeDevice : public Device { in NewDevice() class
40 explicit FakeDevice(const DeviceAttributes& attr) : Device(nullptr, attr) {} in NewDevice() function in tensorflow::__anone296589f0111::NewDevice::FakeDevice
48 return absl::make_unique<FakeDevice>(attr); in NewDevice()
Dsession_mgr_test.cc26 class FakeDevice : public Device { class
28 explicit FakeDevice(const DeviceAttributes& device_attributes) in FakeDevice() function in tensorflow::FakeDevice
40 return std::unique_ptr<Device>(new FakeDevice(device_attributes)); in MakeCPU()
50 FakeDevice::MakeCPU("/job:mnist/replica:0/task:0/device:fakecpu:0")); in SessionMgrTest()
Dcollective_param_resolver_distributed_test.cc36 class FakeDevice : public Device { in NewDevice() class
38 explicit FakeDevice(const DeviceAttributes& attr) : Device(nullptr, attr) {} in NewDevice() function in tensorflow::__anondb24ef490111::NewDevice::FakeDevice
47 return absl::make_unique<FakeDevice>(attr); in NewDevice()
Dcollective_rma_distributed_test.cc47 class FakeDevice : public Device { in NewDevice() class
49 explicit FakeDevice(const DeviceAttributes& attr) : Device(nullptr, attr) {} in NewDevice() function in tensorflow::__anon9c39ea670111::NewDevice::FakeDevice
58 return absl::make_unique<FakeDevice>(attr); in NewDevice()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dmlir_tf2xla.cc47 class FakeDevice : public Device { class
49 explicit FakeDevice(const DeviceAttributes& device_attributes) in FakeDevice() function in tensorflow::__anon6e5af6de0111::FakeDevice
155 FakeDevice device(attr); in ConvertGraphDefToXlaViaMlir()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dplacement_test.cc57 class FakeDevice : public Device { in CreateDevice() class
59 explicit FakeDevice(const DeviceAttributes& attr) : Device(nullptr, attr) {} in CreateDevice() function in tensorflow::__anon751bb52e0111::CreateDevice::FakeDevice
66 return new FakeDevice(attr); in CreateDevice()
Dcontext_test.cc31 class FakeDevice : public Device { in CreateDevice() class
33 explicit FakeDevice(const DeviceAttributes& attr) : Device(nullptr, attr) {} in CreateDevice() function in tensorflow::__anon31761eda0111::CreateDevice::FakeDevice
41 return new FakeDevice(attr); in CreateDevice()
Dtensor_handle_test.cc71 class FakeDevice : public Device { in CreateDevice() class
73 explicit FakeDevice(const DeviceAttributes& attr, bool is_local) in CreateDevice() function in tensorflow::__anonba1fe05b0111::CreateDevice::FakeDevice
90 return new FakeDevice(attr, is_local); in CreateDevice()
/external/tensorflow/tensorflow/compiler/jit/
Dincrease_dynamism_for_auto_jit_pass_test.cc42 class FakeDevice : public Device { class
44 explicit FakeDevice(const DeviceAttributes& device_attributes) in FakeDevice() function in tensorflow::__anon91439fb40111::FakeDevice
55 return absl::make_unique<FakeDevice>(device_attributes); in Make()
65 devices.push_back(FakeDevice::Make(kDeviceName, DEVICE_GPU)); in IncreaseDynamismForAutoJit()
66 devices.push_back(FakeDevice::Make(kHostName, DEVICE_CPU)); in IncreaseDynamismForAutoJit()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Drpc_rendezvous_mgr_test.cc99 class FakeDevice : public Device { in CreateDevice() class
101 explicit FakeDevice(const DeviceAttributes& attr) : Device(nullptr, attr) {} in CreateDevice() function in tensorflow::__anonba1483ce0111::CreateDevice::FakeDevice
108 return new FakeDevice(attr); in CreateDevice()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer.cc113 class FakeDevice : public Device { class
115 FakeDevice(Env* env, const string& device) : Device(env, attr(device)) {} in FakeDevice() function in tensorflow::grappler::__anon36252d010111::FakeDevice
116 explicit FakeDevice(const string& device) : FakeDevice(nullptr, device) {} in FakeDevice() function in tensorflow::grappler::__anon36252d010111::FakeDevice
1386 auto device = absl::make_unique<FakeDevice>(name); in InlineFunctionCalls()