Home
last modified time | relevance | path

Searched full:device_type (Results 1 – 25 of 687) sorted by relevance

12345678910>>...28

/external/tensorflow/tensorflow/dtensor/python/
Dmesh_util.py34 device_type: str, mesh: layout.Mesh) -> None:
36 logging.info('Number of global %s devices: %d', device_type.upper(),
48 device_type: Optional[str] = None
52 if device_type is None:
53 device_type = 'CPU'
56 for d in tf_config.list_logical_devices(device_type)
60 if device_type is None:
61 device_type = devices[0].device_type
63 if device_type.upper() != devices[0].device_type.upper():
65 f'Conflicting devices {str(devices)} and device_type {device_type}')
[all …]
/external/tensorflow/tensorflow/python/framework/
Ddevice_spec.py38 def _as_device_str_or_none(device_type): argument
41 if device_type in ("cpu", "gpu"):
42 return device_type.upper()
43 return _as_str_or_none(device_type)
58 device_spec = DeviceSpec(job="ps", device_type="GPU", device_index=0)
73 device_spec = DeviceSpec(job="ps", device_type="GPU", device_index=0)
86 gpu0_spec = DeviceSpec(job="ps", device_type="GPU", device_index=0)
90 with tf.device(DeviceSpec(device_type="GPU", device_index=1).to_string()):
111 device_type=None, argument
119 device_type: Optional device type string (e.g. "CPU" or "GPU")
[all …]
Ddevice_spec_test.py40 device_type="CPU", device_index=2)
44 self.assertEqual("CPU", d.device_type)
48 d = device_spec_type(device_type="GPU", device_index=0)
58 d.device_type = "CPU"
64 d.device_type = "GPU"
67 d.device_type = "CPU"
70 d.device_type = None
76 device_type="GPU")
88 d = d.replace(device_type="CPU", device_index=0)
94 d = d.replace(device_type="GPU", device_index=2)
[all …]
/external/tensorflow/tensorflow/core/framework/
Ddevice_factory.h34 static void Register(const std::string& device_type,
38 static void Register(const std::string& device_type, DeviceFactory* factory, in Register() argument
40 Register(device_type, std::unique_ptr<DeviceFactory>(factory), priority, in Register()
43 static DeviceFactory* GetFactory(const std::string& device_type);
96 // Return the device priority number for a "device_type" string.
108 static int32 DevicePriority(const std::string& device_type);
110 // Returns true if 'device_type' is registered from plugin. Returns false if
111 // 'device_type' is a first-party device.
112 static bool IsPluggableDevice(const std::string& device_type);
124 // implementation) to use for a specific 'device_type', the
[all …]
Ddevice_factory.cc53 bool IsDeviceFactoryEnabled(const string& device_type) { in IsDeviceFactoryEnabled() argument
62 device_type) != enabled_devices.end(); in IsDeviceFactoryEnabled()
67 int32 DeviceFactory::DevicePriority(const string& device_type) { in DevicePriority() argument
70 auto iter = factories.find(device_type); in DevicePriority()
78 bool DeviceFactory::IsPluggableDevice(const string& device_type) { in IsPluggableDevice() argument
81 auto iter = factories.find(device_type); in IsPluggableDevice()
89 void DeviceFactory::Register(const string& device_type, in Register() argument
92 if (!IsDeviceFactoryEnabled(device_type)) { in Register()
93 LOG(INFO) << "Device factory '" << device_type << "' disabled by " in Register()
99 auto iter = factories.find(device_type); in Register()
[all …]
Dkernel_def_builder_test.cc29 protobuf::TextFormat::ParseFromString("op: 'A' device_type: 'CPU'", in TEST()
42 op: 'B' device_type: 'GPU' in TEST()
56 op: 'C' device_type: 'GPU' in TEST()
68 op: 'D' device_type: 'CPU' in TEST()
84 device_type: 'GPU' in TEST()
104 device_type: 'GPU' in TEST()
126 device_type: 'GPU' in TEST()
145 device_type: 'GPU' in TEST()
167 "op: 'E' device_type: 'GPU' " in TEST()
Dop_kernel_test_base.h73 const DeviceType& device_type,
86 std::unique_ptr<OpKernel> op(CreateOpKernel(device_type, &device,
101 if (dt.first == device_type) {
105 EXPECT_TRUE(found) << "Missing " << device_type << " from "
112 void ExpectFailure(const string& op_type, const DeviceType& device_type, in ExpectFailure() argument
120 std::unique_ptr<OpKernel> op(CreateOpKernel(device_type, &device, in ExpectFailure()
134 EXPECT_NE(dt.first, device_type); in ExpectFailure()
145 const DeviceType& device_type,
156 FindKernelDef(device_type, def, &kernel_def, &kernel_class_name);
/external/mesa3d/.gitlab-ci/
Dlava-gitlab-ci.yml65 --device-type ${DEVICE_TYPE} \
170 DEVICE_TYPE: sun50i-h6-pine-h64
171 DTB: ${DEVICE_TYPE}
182 DEVICE_TYPE: rk3288-veyron-jaq
183 DTB: ${DEVICE_TYPE}
196 DEVICE_TYPE: rk3288-veyron-jaq
197 DTB: ${DEVICE_TYPE}
210 DEVICE_TYPE: rk3399-gru-kevin
211 DTB: ${DEVICE_TYPE}
247 DEVICE_TYPE: meson-g12b-a311d-khadas-vim3
[all …]
/external/autotest/client/common_lib/cros/manual/
Dcfm_helper_unittest.py36 For these testing purposes, we use the device_type ('cameras', etc.)
43 device_type = DEVICE_TYPES[i]
44 mock = self.create_mock_device_getter({device_type: i + 1})
45 setattr(cfm_helper.get_usb_devices, 'get_%s' % device_type, mock)
52 device_type = DEVICE_TYPES[i]
53 self.assertEqual(peripherals[device_type], i + 1)
57 for device_type in DEVICE_TYPES:
58 original_func = getattr(get_usb_devices, 'get_%s' % device_type)
59 setattr(cfm_helper.get_usb_devices, 'get_%s' % device_type,
/external/tensorflow/tensorflow/core/common_runtime/device/
Ddevice_id_manager_test.cc34 DeviceType device_type("GPU"); in TEST() local
37 TF_ASSERT_OK(DeviceIdManager::InsertTfPlatformDeviceIdPair(device_type, key_0, in TEST()
39 EXPECT_EQ(value_0, TfToPlatformDeviceId(device_type, key_0)); in TEST()
42 TF_ASSERT_OK(DeviceIdManager::InsertTfPlatformDeviceIdPair(device_type, key_0, in TEST()
44 EXPECT_EQ(value_0, TfToPlatformDeviceId(device_type, key_0)); in TEST()
49 TF_ASSERT_OK(DeviceIdManager::InsertTfPlatformDeviceIdPair(device_type, key_1, in TEST()
51 EXPECT_EQ(value_1, TfToPlatformDeviceId(device_type, key_1)); in TEST()
55 TF_ASSERT_OK(DeviceIdManager::InsertTfPlatformDeviceIdPair(device_type, key_2, in TEST()
57 EXPECT_EQ(value_1, TfToPlatformDeviceId(device_type, key_2)); in TEST()
61 DeviceIdManager::InsertTfPlatformDeviceIdPair(device_type, key_2, value_0) in TEST()
[all …]
/external/trusty/arm-trusted-firmware/plat/arm/board/tc/fdts/
Dtc_spmc_optee_sp_manifest.dts59 device_type = "cpu";
70 device_type = "cpu";
77 device_type = "cpu";
84 device_type = "cpu";
91 device_type = "cpu";
98 device_type = "cpu";
105 device_type = "cpu";
112 device_type = "cpu";
120 device_type = "memory";
125 device_type = "ns-memory";
Dtc_spmc_manifest.dts60 device_type = "cpu";
71 device_type = "cpu";
78 device_type = "cpu";
85 device_type = "cpu";
92 device_type = "cpu";
99 device_type = "cpu";
106 device_type = "cpu";
113 device_type = "cpu";
121 device_type = "memory";
128 device_type = "ns-memory";
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dcustom_graph_optimizer_registry.cc36 // device_type as its key, and an optimizer creator as the value.
46 // device_type as its key, and ConfigList as the value.
114 LOG(INFO) << "Plugin optimizer for device_type " << it->first in CreateOptimizers()
123 const Creator& optimizer_creator, const std::string& device_type, in RegisterPluginOptimizerOrDie() argument
125 auto ret = GetPluginConfigMap()->insert({device_type, configs}); in RegisterPluginOptimizerOrDie()
127 LOG(FATAL) << "PluginGraphOptimizer with device_type " // Crash OK in RegisterPluginOptimizerOrDie()
128 << device_type << " is registered twice."; in RegisterPluginOptimizerOrDie()
130 GetPluginRegistrationMap()->insert({device_type, optimizer_creator}); in RegisterPluginOptimizerOrDie()
138 for (const auto& device_type : device_types) { in PrintPluginConfigsIfConflict() local
139 const auto it = GetPluginConfigMap()->find(device_type); in PrintPluginConfigsIfConflict()
[all …]
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_test_util.cc143 int num_workers, int num_devices_per_worker, DeviceType device_type) { in CreateCollectiveTestEnv() argument
152 test_env->device_type = device_type; in CreateCollectiveTestEnv()
155 if (device_type == DEVICE_CPU) { in CreateCollectiveTestEnv()
158 } else if (device_type == DEVICE_GPU) { in CreateCollectiveTestEnv()
166 LOG(FATAL) << "Unsupported device_type " << device_type; in CreateCollectiveTestEnv()
216 col_params->group.device_type = test_env.device_type; in CreateCollectiveParams()
224 task_name, "/device:", test_env.device_type.type_string(), ":", di)); in CreateCollectiveParams()
263 if (device->device_type() == DEVICE_CPU) { in CopyTensorToDevice()
265 } else if (device->device_type() == DEVICE_GPU) { in CopyTensorToDevice()
274 LOG(FATAL) << "Unsupported device_type " << device->device_type(); in CopyTensorToDevice()
[all …]
Dmemory_types.h25 // 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,
/external/arm-trusted-firmware/plat/arm/board/tc/fdts/
Dtc_spmc_manifest.dts60 device_type = "cpu";
71 device_type = "cpu";
78 device_type = "cpu";
85 device_type = "cpu";
92 device_type = "cpu";
99 device_type = "cpu";
106 device_type = "cpu";
113 device_type = "cpu";
122 device_type = "memory";
Dtc_spmc_optee_sp_manifest.dts59 device_type = "cpu";
70 device_type = "cpu";
77 device_type = "cpu";
84 device_type = "cpu";
91 device_type = "cpu";
98 device_type = "cpu";
105 device_type = "cpu";
112 device_type = "cpu";
121 device_type = "memory";
/external/mesa3d/src/panfrost/ci/
Dgitlab-ci.yml99 # https://lava.collabora.dev/scheduler/device_type/meson-g12b-a311d-khadas-vim3
102 DEVICE_TYPE: meson-g12b-a311d-khadas-vim3
103 DTB: $DEVICE_TYPE
109 # Status: https://lava.collabora.dev/scheduler/device_type/mt8183-kukui-jacuzzi-juniper-sku16
112 DEVICE_TYPE: mt8183-kukui-jacuzzi-juniper-sku16
113 DTB: $DEVICE_TYPE
120 # Status: https://lava.collabora.dev/scheduler/device_type/mt8192-asurada-spherion-r0
124 DEVICE_TYPE: mt8192-asurada-spherion-r0
125 DTB: $DEVICE_TYPE
134 DEVICE_TYPE: rk3288-veyron-jaq
[all …]
/external/autotest/server/site_tests/bluetooth_AdapterSRHealth/
Dbluetooth_AdapterSRHealth.py81 * device_type: MOUSE, BLE_MOUSE, etc.
132 for device_type, device, _ in devtuples:
133 if 'BLE' in device_type:
155 device_type = 'MOUSE'
156 device = self.devices[device_type][0]
157 self.run_reconnect_device([(device_type, device,
163 device_type = 'BLE_MOUSE'
164 device = self.devices[device_type][0]
165 self.run_reconnect_device([(device_type, device,
214 device_type = 'MOUSE'
[all …]
/external/angle/scripts/
Dupdate_extension_data.py108 def name_device(gpu, device_type): argument
112 assert device_type
113 return DEVICE_NAME_MAP[device_type]
124 def get_props_string(gpu, bot_os, device_os, device_type): argument
125 d = {'gpu': gpu, 'os': bot_os, 'device os': device_os, 'device': device_type}
133 device_type = found_dims.get('device_type', None)
135 (task_id, get_props_string(gpu, bot_os, device_os, device_type)))
136 target_file_name = '%s_%s.json' % (name_device(gpu, device_type), name_os(bot_os, device_os))
197 'device_type': DEVICES_TYPES,
210 found_gpu_or_device = ('gpu' in found_dims or 'device_type' in found_dims)
/external/mesa3d/src/intel/ci/
Dgitlab-ci-inc.yml185 # Status: https://lava.collabora.dev/scheduler/device_type/acer-cb317-1h-c3z6-dedede
188 DEVICE_TYPE: acer-cb317-1h-c3z6-dedede
194 # Status: https://lava.collabora.dev/scheduler/device_type/asus-C523NA-A20057-coral
197 DEVICE_TYPE: asus-C523NA-A20057-coral
203 # Status: https://lava.collabora.dev/scheduler/device_type/hp-x360-12b-ca0010nr-n4020-octopus
206 DEVICE_TYPE: hp-x360-12b-ca0010nr-n4020-octopus
212 # Status: https://lava.collabora.dev/scheduler/device_type/asus-cx9400-volteer
216 DEVICE_TYPE: asus-cx9400-volteer
222 # Status: https://lava.collabora.dev/scheduler/device_type/asus-C433TA-AJ0005-rammus
225 DEVICE_TYPE: asus-C433TA-AJ0005-rammus
[all …]
/external/rust/crates/virtio-drivers/src/transport/
Dpci.rs53 fn device_type(pci_device_id: u16) -> DeviceType { in device_type() function
71 let device_type = device_type(device_function_info.device_id); in virtio_device_type() localVariable
72 if device_type != DeviceType::Invalid { in virtio_device_type()
73 return Some(device_type); in virtio_device_type()
84 device_type: DeviceType, field
113 let device_type = device_type(device_id); in new() localVariable
191 device_type, in new()
203 fn device_type(&self) -> DeviceType { in device_type() method
204 self.device_type in device_type()
521 assert_eq!(device_type(0x1000), DeviceType::Network); in transitional_device_ids()
[all …]
/external/cronet/base/system/
Dsystem_monitor.cc33 void SystemMonitor::ProcessDevicesChanged(DeviceType device_type) { in ProcessDevicesChanged() argument
34 NotifyDevicesChanged(device_type); in ProcessDevicesChanged()
45 void SystemMonitor::NotifyDevicesChanged(DeviceType device_type) { in NotifyDevicesChanged() argument
46 DVLOG(1) << "DevicesChanged with device type " << device_type; in NotifyDevicesChanged()
48 FROM_HERE, &DevicesChangedObserver::OnDevicesChanged, device_type); in NotifyDevicesChanged()
/external/cronet/build/fuchsia/
Dupdate_images.py109 device_type, arch = components
110 device_images = all_device_types if device_type == '*' else [device_type]
119 device_type = image_to_download[0]
121 image_output_dir = os.path.join(image_root_dir, arch, device_type)
125 logging.info('Downloading Fuchsia boot images for %s.%s...', device_type,
131 device_type not in legacy_delimiter_device_types:
137 '{device_type}{type_arch_connector}{arch}.tgz'.format(
138 bucket=bucket, image_hash=image_hash, device_type=device_type,
/external/libchrome/base/system_monitor/
Dsystem_monitor.cc33 void SystemMonitor::ProcessDevicesChanged(DeviceType device_type) { in ProcessDevicesChanged() argument
34 NotifyDevicesChanged(device_type); in ProcessDevicesChanged()
45 void SystemMonitor::NotifyDevicesChanged(DeviceType device_type) { in NotifyDevicesChanged() argument
46 DVLOG(1) << "DevicesChanged with device type " << device_type; in NotifyDevicesChanged()
48 FROM_HERE, &DevicesChangedObserver::OnDevicesChanged, device_type); in NotifyDevicesChanged()

12345678910>>...28