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::__anon08ffc4550111::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::__anon0a0fc39b0111::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::__anonc326b8990111::CreateDevice::FakeDevice
Dbuf_rendezvous_test.cc34 class FakeDevice : public Device { in NewDevice() class
36 explicit FakeDevice(const DeviceAttributes& attrs) in NewDevice() function in tensorflow::__anon7fe7a2110111::BufRendezvousTest::NewDevice::FakeDevice
Dconstant_folding_test.cc96 class FakeDevice : public Device { class
98 explicit FakeDevice(const DeviceAttributes& device_attributes) in FakeDevice() function in tensorflow::__anon0d16402c0111::FakeDevice
Dplacer_test.cc85 class FakeDevice : public Device { class
87 explicit FakeDevice(const DeviceAttributes& device_attributes) in FakeDevice() function in tensorflow::__anonced950570111::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::__anone11ecd1a0111::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::__anon69b9b93c0111::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::__anonbc28e37d0111::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::__anonf35127a70111::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::__anonba1791850111::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::__anone67ae04c0111::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::__anonfc33ea780111::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::__anoncf3b6a790111::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::__anon48e36a890111::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::__anon75af4bec0111::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::__anonc0376b920111::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::__anond52f36620111::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::__anon36c8eb5f0111::FakeDevice
116 explicit FakeDevice(const string& device) : FakeDevice(nullptr, device) {} in FakeDevice() function in tensorflow::grappler::__anon36c8eb5f0111::FakeDevice