| /external/tensorflow/tensorflow/core/common_runtime/ |
| memory_types.h | 25 // Returns an error iff *g running on a single device of 'device_type'
27 Status ValidateMemoryTypes(const DeviceType& device_type, const Graph* g);
31 // Send/HostRecv nodes. 'device_type' specifies the type of device on
38 Status EnsureMemoryTypes(const DeviceType& device_type,
42 // running on 'device_type'.
43 Status MemoryTypeForOutput(const DeviceType& device_type, const Graph* g,
|
| device_factory.h | 33 static void Register(const string& device_type, DeviceFactory* factory,
35 static DeviceFactory* GetFactory(const string& device_type);
56 // Return the device priority number for a "device_type" string.
68 static int32 DevicePriority(const string& device_type);
80 // implementation) to use for a specific 'device_type', the
90 // 2) When choosing which 'device_type' is preferred over other
107 explicit Registrar(const string& device_type, int priority = 50) {
108 DeviceFactory::Register(device_type, new Factory(), priority);
114 #define REGISTER_LOCAL_DEVICE_FACTORY(device_type, device_factory, ...) \
115 INTERNAL_REGISTER_LOCAL_DEVICE_FACTORY(device_type, device_factory,
[all...] |
| device_factory.cc | 53 int32 DeviceFactory::DevicePriority(const string& device_type) {
56 auto iter = factories.find(device_type);
65 void DeviceFactory::Register(const string& device_type, DeviceFactory* factory,
70 auto iter = factories.find(device_type);
72 factories[device_type] = {std::move(factory_ptr), priority};
78 << device_type << " with the same priority " << priority;
83 DeviceFactory* DeviceFactory::GetFactory(const string& device_type) {
85 auto it = device_factories().find(device_type);
|
| memory_types.cc | 48 const DeviceType& device_type, const Graph* g,
50 if (device_type != DEVICE_GPU && device_type != DEVICE_SYCL) {
66 TF_RETURN_IF_ERROR(MemoryTypesForNode(g->op_registry(), device_type,
93 Status ValidateMemoryTypes(const DeviceType& device_type, const Graph* g) {
95 device_type, g, [g](const Edge* e, MemoryType sm, MemoryType dm) {
149 Status EnsureMemoryTypes(const DeviceType& device_type,
158 device_type, g, [g, &edges](const Edge* e, MemoryType sm, MemoryType dm) {
201 return ValidateMemoryTypes(device_type, g);
204 Status MemoryTypeForOutput(const DeviceType& device_type, const Graph* g
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| memory_types.h | 32 const DeviceType& device_type, const NodeDef& ndef,
|
| kernel_def_builder.h | 39 KernelDefBuilder& Device(const char* device_type);
40 // KernelDefBuilder& Device(DeviceType device_type);
|
| kernel_def_builder.cc | 32 KernelDefBuilder& KernelDefBuilder::Device(const char* device_type) {
33 kernel_def_->set_device_type(device_type);
|
| /device/google/cuttlefish/shared/config/ |
| gen_wpa_supplicant_conf.sh | 19 device_type=10-0050F204-5
|
| /external/tensorflow/tensorflow/python/client/ |
| device_lib_test.py | 32 self.assertEqual(devices[0].device_type, "CPU")
37 self.assertTrue("GPU" in [d.device_type for d in devices] or
38 "SYCL" in [d.device_type for d in devices])
|
| /external/autotest/server/site_tests/enterprise_CFM_USBPeripheralHotplugStress/ |
| enterprise_CFM_USBPeripheralHotplugStress.py | 62 def _check_peripheral(self, device_type, hub_on, peripheral_dict,
67 @param device_type: The type of the peripheral.
78 device_name = peripheral_dict.get(device_type)
84 raise error.TestFail('%s not detected.' % device_type)
87 raise error.TestFail('%s should not be detected.' % device_type)
91 raise error.TestFail('Available %s not selected.' % device_type)
94 raise error.TestFail('Available %s not selected.' % device_type)
97 logging.info("[SUCCESS] %s has been detected.", device_type)
99 logging.info("[SUCCESS] %s has not been detected.", device_type)
|
| /external/tensorflow/tensorflow/python/framework/ |
| device.py | 37 device_spec = DeviceSpec(job="ps", device_type="GPU", device_index=0)
51 with tf.device(DeviceSpec(job="ps", device_type="GPU", device_index=0):
53 with tf.device(DeviceSpec(device_type="GPU", device_index=1):
67 def __init__(self, job=None, replica=None, task=None, device_type=None,
75 device_type: Optional device type string (e.g. "CPU" or "GPU")
82 if device_type == "cpu" or device_type == "gpu":
85 self.device_type = device_type.upper()
87 self.device_type = device_typ
[all...] |
| device_test.py | 36 device_type="CPU", device_index=2)
40 self.assertEqual("CPU", d.device_type)
44 d = device.DeviceSpec(device_type="GPU", device_index=0)
53 d.device_type = "CPU"
59 d.device_type = "GPU"
62 d.device_type = "CPU"
65 d.device_type = None
71 d = device.DeviceSpec(job="foo", replica=12, task=3, device_type="GPU")
|
| /external/tensorflow/tensorflow/core/kernels/ |
| ops_testutil.cc | 25 void OpsTestBase::SetDevice(const DeviceType& device_type,
28 device_type_ = device_type;
31 if (device_type == DEVICE_GPU) {
39 CHECK_NE(device_type, DEVICE_GPU)
|
| /device/linaro/bootloader/OpenPlatformPkg/Platforms/ARM/VExpress/DeviceTree/ |
| fvp-base-gicv2-psci.dts | 125 device_type = "cpu";
133 device_type = "cpu";
141 device_type = "cpu";
149 device_type = "cpu";
157 device_type = "cpu";
165 device_type = "cpu";
173 device_type = "cpu";
181 device_type = "cpu";
190 device_type = "memory";
|
| fvp-base-gicv2legacy-psci.dts | 125 device_type = "cpu";
133 device_type = "cpu";
141 device_type = "cpu";
149 device_type = "cpu";
157 device_type = "cpu";
165 device_type = "cpu";
173 device_type = "cpu";
181 device_type = "cpu";
190 device_type = "memory";
|
| fvp-base-gicv3-psci.dts | 123 device_type = "cpu";
131 device_type = "cpu";
139 device_type = "cpu";
147 device_type = "cpu";
155 device_type = "cpu";
163 device_type = "cpu";
171 device_type = "cpu";
179 device_type = "cpu";
188 device_type = "memory";
|
| fvp-foundation-gicv2-psci.dts | 110 device_type = "cpu";
118 device_type = "cpu";
126 device_type = "cpu";
134 device_type = "cpu";
143 device_type = "memory";
|
| fvp-foundation-gicv2legacy-psci.dts | 110 device_type = "cpu";
118 device_type = "cpu";
126 device_type = "cpu";
134 device_type = "cpu";
143 device_type = "memory";
|
| fvp-foundation-gicv3-psci.dts | 108 device_type = "cpu";
116 device_type = "cpu";
124 device_type = "cpu";
132 device_type = "cpu";
141 device_type = "memory";
|
| /device/linaro/bootloader/arm-trusted-firmware/fdts/ |
| fvp-base-gicv2-psci-aarch32.dts | 99 device_type = "cpu";
108 device_type = "cpu";
117 device_type = "cpu";
126 device_type = "cpu";
135 device_type = "cpu";
144 device_type = "cpu";
153 device_type = "cpu";
162 device_type = "cpu";
176 device_type = "memory";
|
| fvp-base-gicv2-psci.dts | 99 device_type = "cpu";
108 device_type = "cpu";
117 device_type = "cpu";
126 device_type = "cpu";
135 device_type = "cpu";
144 device_type = "cpu";
153 device_type = "cpu";
162 device_type = "cpu";
176 device_type = "memory";
|
| fvp-base-gicv3-psci-aarch32.dts | 99 device_type = "cpu";
108 device_type = "cpu";
117 device_type = "cpu";
126 device_type = "cpu";
135 device_type = "cpu";
144 device_type = "cpu";
153 device_type = "cpu";
162 device_type = "cpu";
176 device_type = "memory";
|
| fvp-base-gicv3-psci-common.dtsi | 97 device_type = "cpu";
106 device_type = "cpu";
115 device_type = "cpu";
124 device_type = "cpu";
133 device_type = "cpu";
142 device_type = "cpu";
151 device_type = "cpu";
160 device_type = "cpu";
174 device_type = "memory";
|
| fvp-foundation-gicv2-psci.dts | 84 device_type = "cpu";
93 device_type = "cpu";
102 device_type = "cpu";
111 device_type = "cpu";
125 device_type = "memory";
|
| /external/autotest/client/site_tests/bluetooth_IDCheck/ |
| bluetooth_IDCheck.py | 56 device_type = utils.get_device_type().lower()
57 alias_format = '%s_[a-z0-9]{4}' % device_type
|
