| /external/u-boot/arch/arm/dts/ |
| D | thunderx-88xx.dtsi | 25 device_type = "cpu";
31 device_type = "cpu";
37 device_type = "cpu";
43 device_type = "cpu";
49 device_type = "cpu";
55 device_type = "cpu";
61 device_type = "cpu";
67 device_type = "cpu";
73 device_type = "cpu";
79 device_type = "cpu";
[all …]
|
| D | armada-xp-mv78460.dtsi | 69 device_type = "cpu";
77 device_type = "cpu";
85 device_type = "cpu";
93 device_type = "cpu";
110 device_type = "pci";
154 device_type = "pci";
171 device_type = "pci";
188 device_type = "pci";
205 device_type = "pci";
222 device_type = "pci";
[all …]
|
| D | hi6220.dtsi | 58 device_type = "cpu";
65 device_type = "cpu";
72 device_type = "cpu";
79 device_type = "cpu";
86 device_type = "cpu";
93 device_type = "cpu";
100 device_type = "cpu";
107 device_type = "cpu";
|
| D | r8a7795-h3ulcb.dts | 18 device_type = "memory";
24 device_type = "memory";
29 device_type = "memory";
34 device_type = "memory";
|
| D | r8a7795-salvator-x.dts | 17 device_type = "memory";
23 device_type = "memory";
28 device_type = "memory";
33 device_type = "memory";
|
| D | armada-xp-mv78260.dtsi | 68 device_type = "cpu";
76 device_type = "cpu";
93 device_type = "pci";
133 device_type = "pci";
150 device_type = "pci";
167 device_type = "pci";
184 device_type = "pci";
201 device_type = "pci";
218 device_type = "pci";
235 device_type = "pci";
[all …]
|
| D | fsl-ls2080a.dtsi | 15 device_type = "memory";
38 device_type = "serial";
46 device_type = "serial";
100 device_type = "pci";
115 device_type = "pci";
130 device_type = "pci";
145 device_type = "pci";
|
| D | keystone-k2e.dtsi | 20 device_type = "cpu";
26 device_type = "cpu";
32 device_type = "cpu";
38 device_type = "cpu";
100 device_type = "pci";
|
| D | bcm2836.dtsi | 42 device_type = "cpu";
49 device_type = "cpu";
56 device_type = "cpu";
63 device_type = "cpu";
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | device.py | 68 def __init__(self, job=None, replica=None, task=None, device_type=None, argument
83 if device_type == "cpu" or device_type == "gpu":
86 self.device_type = device_type.upper()
88 self.device_type = device_type
96 self.device_type = None
163 if self.device_type is not None:
165 self.device_type = y[0].upper()
169 if self.device_type is not None:
171 self.device_type = y[1]
191 if dev.device_type is not None:
[all …]
|
| D | device_test.py | 39 device_type="CPU", device_index=2)
43 self.assertEqual("CPU", d.device_type)
47 d = device.DeviceSpec(device_type="GPU", device_index=0)
56 d.device_type = "CPU"
62 d.device_type = "GPU"
65 d.device_type = "CPU"
68 d.device_type = None
74 d = device.DeviceSpec(job="foo", replica=12, task=3, device_type="GPU")
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | device_factory.h | 33 static void Register(const string& device_type, DeviceFactory* factory,
35 static DeviceFactory* GetFactory(const string& device_type);
69 static int32 DevicePriority(const string& device_type);
108 explicit Registrar(const string& device_type, int priority = 50) {
109 DeviceFactory::Register(device_type, new Factory(), priority);
115 #define REGISTER_LOCAL_DEVICE_FACTORY(device_type, device_factory, ...) \ argument
116 INTERNAL_REGISTER_LOCAL_DEVICE_FACTORY(device_type, device_factory, \
119 #define INTERNAL_REGISTER_LOCAL_DEVICE_FACTORY(device_type, device_factory, \ argument
122 INTERNAL_REGISTER_LOCAL_DEVICE_FACTORY_NAME(ctr)(device_type, \
|
| D | device_factory.cc | 54 int32 DeviceFactory::DevicePriority(const string& device_type) { in DevicePriority() argument
57 auto iter = factories.find(device_type); in DevicePriority()
66 void DeviceFactory::Register(const string& device_type, DeviceFactory* factory, in Register() argument
71 auto iter = factories.find(device_type); in Register()
73 factories[device_type] = {std::move(factory_ptr), priority}; in Register()
79 << device_type << " with the same priority " << priority; in Register()
84 DeviceFactory* DeviceFactory::GetFactory(const string& device_type) { in GetFactory() argument
86 auto it = device_factories().find(device_type); in GetFactory()
|
| D | memory_types.cc | 48 const DeviceType& device_type, const Graph* g, in ProcessMemoryTypes() argument
50 if (device_type != DEVICE_GPU && device_type != DEVICE_SYCL) { in ProcessMemoryTypes()
66 TF_RETURN_IF_ERROR(MemoryTypesForNode(g->op_registry(), device_type, in ProcessMemoryTypes()
93 Status ValidateMemoryTypes(const DeviceType& device_type, const Graph* g) { in ValidateMemoryTypes() argument
95 device_type, g, [](const Edge* e, MemoryType sm, MemoryType dm) { in ValidateMemoryTypes()
150 Status EnsureMemoryTypes(const DeviceType& device_type, in EnsureMemoryTypes() argument
159 device_type, g, [&edges](const Edge* e, MemoryType sm, MemoryType dm) { in EnsureMemoryTypes()
202 return ValidateMemoryTypes(device_type, g); in EnsureMemoryTypes()
205 Status MemoryTypeForOutput(const DeviceType& device_type, const Graph* g, in MemoryTypeForOutput() argument
209 TF_RETURN_IF_ERROR(MemoryTypesForNode(g->op_registry(), device_type, n->def(), in MemoryTypeForOutput()
|
| D | ring_reducer_test.cc | 98 const DeviceType& device_type, in GetKernel() argument
102 device_type, device, device->GetAllocator(AllocatorAttributes()), node, in GetKernel()
110 std::unique_ptr<OpKernel> GetAdd(DataType dtype, const DeviceType& device_type, in GetAdd() argument
118 return GetKernel(node_def, device_type, device); in GetAdd()
121 std::unique_ptr<OpKernel> GetDiv(DataType dtype, const DeviceType& device_type, in GetDiv() argument
129 return GetKernel(node_def, device_type, device); in GetDiv()
156 const DeviceType& device_type, int num_subdivs, int fail_after) { in Init() argument
160 device_type_ = device_type; in Init()
168 if (device_type == DEVICE_CPU) { in Init()
173 } else if (device_type == DEVICE_GPU && !gpu_devices_.empty()) { in Init()
[all …]
|
| D | hierarchical_tree_broadcaster_test.cc | 213 const DeviceType& device_type, int fail_after) { in Init() argument
220 device_type_ = device_type; in Init()
228 if (device_type == DEVICE_CPU) { in Init()
233 } else if (device_type == DEVICE_GPU && !gpu_devices_.empty()) { in Init()
242 LOG(FATAL) << "Unsupported device_type " << device_type; in Init()
246 if (!dev_mgr_ || device_type == DEVICE_CPU) { in Init()
261 col_params_.group.device_type = device_type; in Init()
320 if (device_type == DEVICE_GPU) { in Init()
336 device_type, this)); in Init()
359 const DeviceType& device_type, in GetKernel() argument
[all …]
|
| D | ring_gatherer_test.cc | 98 const DeviceType& device_type, in GetKernel() argument
102 device_type, device, device->GetAllocator(AllocatorAttributes()), node, in GetKernel()
134 const DeviceType& device_type, int num_subdivs, int fail_after) { in Init() argument
138 device_type_ = device_type; in Init()
146 if (device_type == DEVICE_CPU) { in Init()
151 } else if (device_type == DEVICE_GPU && !gpu_devices_.empty()) { in Init()
160 LOG(FATAL) << "Unsupported device_type " << device_type; in Init()
164 if (!dev_mgr_ || device_type == DEVICE_CPU) { in Init()
178 col_params_.group.device_type = device_type; in Init()
216 if (device_type == DEVICE_GPU) { in Init()
[all …]
|
| D | memory_types.h | 27 Status ValidateMemoryTypes(const DeviceType& device_type, const Graph* g);
38 Status EnsureMemoryTypes(const DeviceType& device_type,
43 Status MemoryTypeForOutput(const DeviceType& device_type, const Graph* g,
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | partially_decluster_pass.cc | 58 DeviceType device_type(""); in FindNodesToDecluster() local
60 DeviceToDeviceType(n->assigned_device_name(), &device_type)); in FindNodesToDecluster()
61 TF_RETURN_IF_ERROR(MemoryTypesForNode(graph.op_registry(), device_type, in FindNodesToDecluster()
82 TF_RETURN_IF_ERROR(MemoryTypesForNode(graph.op_registry(), device_type, in FindNodesToDecluster()
227 bool IsMustCompileDevice(const DeviceType& device_type) { in IsMustCompileDevice() argument
229 if (XlaOpRegistry::GetCompilationDevice(device_type.type(), ®istration)) { in IsMustCompileDevice()
238 DeviceType device_type(""); in MustCompileNode() local
240 DeviceToDeviceType(n->assigned_device_name(), &device_type)); in MustCompileNode()
242 if (IsMustCompileDevice(device_type)) { in MustCompileNode()
248 *must_compile = !FindKernelDef(device_type, n->def(), nullptr, nullptr).ok(); in MustCompileNode()
|
| D | mark_for_compilation_pass.cc | 490 DeviceType device_type(""); in FindCompilationCandidates() local
492 DeviceToDeviceType(node->assigned_device_name(), &device_type)); in FindCompilationCandidates()
494 << device_type.type_string(); in FindCompilationCandidates()
496 if (is_compilable_fn && !is_compilable_fn(node, device_type)) { in FindCompilationCandidates()
503 XlaOpRegistry::GetCompilationDevice(device_type.type(), ®istration)); in FindCompilationCandidates()
656 CHECK(XlaOpRegistry::GetCompilationDevice(device->device_type(), in IsCompilable()
714 auto is_compilable = [&](const Node* node, const DeviceType& device_type) { in Run() argument
716 if (!XlaOpRegistry::GetCompilationDevice(device_type.type(), in Run()
931 DeviceType device_type(""); in IgnoreResourceOpForSafetyAnalysis() local
933 DeviceToDeviceType(n.assigned_device_name(), &device_type)); in IgnoreResourceOpForSafetyAnalysis()
[all …]
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | device_util.py | 48 assert d.device_type is None or d.device_type == d.device_type.upper(), (
49 "Device type '%s' must be all-caps." % (d.device_type,))
52 replica=0, task=0, device_type="CPU", device_index=0)
|
| /external/tensorflow/tensorflow/python/client/ |
| D | device_lib_test.py | 33 self.assertEqual(devices[0].device_type, "CPU")
37 self.assertEqual(devices[0].device_type, "CPU")
42 self.assertTrue("GPU" in [d.device_type for d in devices] or
43 "SYCL" in [d.device_type for d in devices])
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ops_testutil.cc | 25 void OpsTestBase::SetDevice(const DeviceType& device_type, in SetDevice() argument
28 device_type_ = device_type; in SetDevice()
31 if (device_type == DEVICE_GPU) { in SetDevice()
39 CHECK_NE(device_type, DEVICE_GPU) in SetDevice()
|
| /external/u-boot/arch/x86/cpu/ |
| D | cpu_x86.c | 67 const char *device_type; in cpu_x86_get_count() local
69 device_type = fdt_getprop(gd->fdt_blob, cpu, in cpu_x86_get_count()
71 if (!device_type) in cpu_x86_get_count()
73 if (strcmp(device_type, "cpu") == 0) in cpu_x86_get_count()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.cc | 118 expensive_ = context->device_type() != DeviceType(DEVICE_GPU) && in OpKernel()
119 context->device_type() != DeviceType(DEVICE_SYCL); in OpKernel()
202 DeviceType device_type, DeviceBase* device, Allocator* allocator, in OpKernelConstruction() argument
208 : device_type_(std::move(device_type)), in OpKernelConstruction()
1060 static string Key(StringPiece op_type, const DeviceType& device_type, in Key() argument
1062 return strings::StrCat(op_type, ":", DeviceTypeString(device_type), ":", in Key()
1074 Key(kernel_def->op(), DeviceType(kernel_def->device_type()), in InitInternal()
1103 Status FindKernelRegistration(const DeviceType& device_type, in FindKernelRegistration() argument
1112 const string key = Key(node_def.op(), device_type, label); in FindKernelRegistration()
1137 bool KernelDefAvailable(const DeviceType& device_type, in KernelDefAvailable() argument
[all …]
|