Home
last modified time | relevance | path

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

/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::__anondeda8fc40111::CreateDevice::FakeDevice
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::__anondfea8f0a0111::Dev::FakeDevice
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::__anon029183080111::CreateDevice::FakeDevice
Dbuf_rendezvous_test.cc34 class FakeDevice : public Device { in NewDevice() class
36 explicit FakeDevice(const DeviceAttributes& attrs) in NewDevice() function in tensorflow::__anon14088d000111::BufRendezvousTest::NewDevice::FakeDevice
Dconstant_folding_test.cc96 class FakeDevice : public Device { class
98 explicit FakeDevice(const DeviceAttributes& device_attributes) in FakeDevice() function in tensorflow::__anon2d1da2db0111::FakeDevice
Dplacer_test.cc85 class FakeDevice : public Device { class
87 explicit FakeDevice(const DeviceAttributes& device_attributes) in FakeDevice() function in tensorflow::__anon6791bc460111::FakeDevice
/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::__anon208997890111::FakeDevice
/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::__anonca9385ab0111::FakeDevice
/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::__anond5a5c42c0111::NewDevice::FakeDevice
Dsession_mgr_test.cc29 class FakeDevice : public Device { class
31 explicit FakeDevice(const DeviceAttributes& device_attributes) in FakeDevice() function in tensorflow::FakeDevice
Dcollective_param_resolver_distributed_test.cc39 class FakeDevice : public Device { in NewDevice() class
41 explicit FakeDevice(const DeviceAttributes& attr) : Device(nullptr, attr) {} in NewDevice() function in tensorflow::__anon60c5ef160111::NewDevice::FakeDevice
Dcollective_rma_distributed_test.cc59 class FakeDevice : public Device { in NewDevice() class
61 explicit FakeDevice(const DeviceAttributes& attr, Allocator* allocator) in NewDevice() function in tensorflow::__anon941398540111::NewDevice::FakeDevice
/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::__anonfeb9291b0111::CreateDevice::FakeDevice
Dcontext_test.cc35 class FakeDevice : public Device { in CreateDevice() class
37 explicit FakeDevice(const DeviceAttributes& attr) : Device(nullptr, attr) {} in CreateDevice() function in tensorflow::__anonab5bb3870111::CreateDevice::FakeDevice
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::__anonfbff82c80111::CreateDevice::FakeDevice
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Ddevice_util_test.cc44 class FakeDevice : public Device { class
46 explicit FakeDevice(const DeviceAttributes& device_attributes) in FakeDevice() function in tensorflow::__anon7c7bd4780111::FakeDevice
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Drpc_rendezvous_mgr_test.cc103 class FakeDevice : public Device { in CreateDevice() class
105 explicit FakeDevice(const DeviceAttributes& attr) : Device(nullptr, attr) {} in CreateDevice() function in tensorflow::__anonc372433b0111::CreateDevice::FakeDevice
/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::__anonc6594cc10111::FakeDevice
/external/tensorflow/tensorflow/core/tfrt/eager/cpp_tests/core_runtime/
Dop_handler_selector_test.cc95 class FakeDevice : public tensorflow::Device { in CreateDevice() class
97 explicit FakeDevice(const DeviceAttributes& attr) : Device(nullptr, attr) {} in CreateDevice() function in tfrt::tf::__anonafe39e510111::CreateDevice::FakeDevice
/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::__anon7633b5ce0111::FakeDevice
116 explicit FakeDevice(const string& device) : FakeDevice(nullptr, device) {} in FakeDevice() function in tensorflow::grappler::__anon7633b5ce0111::FakeDevice