Home
last modified time | relevance | path

Searched refs:MakeDevice (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dcomposite_device_test.cc30 CompositeDevice::MakeDevice(underlying_devices, /*unique_device_id=*/0, in TEST()
46 CompositeDevice::MakeDevice(underlying_devices, /*unique_device_id=*/0, in TEST()
58 CompositeDevice::MakeDevice(underlying_devices, /*unique_device_id=*/1, in TEST()
76 CompositeDevice::MakeDevice(underlying_devices, composite_device_name, in TEST()
Dcomposite_device.cc25 std::unique_ptr<CompositeDevice> CompositeDevice::MakeDevice( in MakeDevice() function in tensorflow::CompositeDevice
32 return CompositeDevice::MakeDevice(underlying_devices, device_name, status); in MakeDevice()
35 std::unique_ptr<CompositeDevice> CompositeDevice::MakeDevice( in MakeDevice() function in tensorflow::CompositeDevice
Dcomposite_device.h47 static std::unique_ptr<CompositeDevice> MakeDevice(
52 static std::unique_ptr<CompositeDevice> MakeDevice(
Dplacer_test.cc95 static std::unique_ptr<Device> MakeDevice(const string& name, in MakeDevice() function in tensorflow::__anon24cf05320111::FakeDevice
104 return MakeDevice(name, "FakeCPU"); in MakeCPU()
108 return MakeDevice(name, "FakeGPU"); in MakeGPU()
233 local_devices_.emplace_back(FakeDevice::MakeDevice( in PlacerTest()
236 local_devices_.emplace_back(FakeDevice::MakeDevice( in PlacerTest()
Dprocess_function_library_runtime_test.cc861 CompositeDevice::MakeDevice({device0_->name(), device1_->name()}, in TEST_F()
1197 CompositeDevice::MakeDevice({device0_->name(), device1_->name()}, in TEST_F()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute_node_test.cc63 CompositeDevice::MakeDevice({device0->name(), device1->name()}, in TEST()
Dcontext.cc940 device = CompositeDevice::MakeDevice(underlying_devices, in FindOrCreateCompositeDevice()
944 device = CompositeDevice::MakeDevice(underlying_devices, device_name, &s); in FindOrCreateCompositeDevice()