| /external/tensorflow/tensorflow/python/framework/ |
| D | config.py | 307 def list_physical_devices(device_type=None): argument
342 def list_logical_devices(device_type=None): argument
379 def get_visible_devices(device_type=None): argument
413 def set_visible_devices(devices, device_type=None): argument
|
| D | device_spec.py | 42 def _as_device_str_or_none(device_type): argument
111 def __init__(self, job=None, replica=None, task=None, device_type=None, argument
273 def device_type(self): member in DeviceSpecV2
351 def _components_to_string(job, replica, task, device_type, device_index): argument
417 def device_type(self, device_type): member in DeviceSpecV1
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | device_factory.h | 124 #define REGISTER_LOCAL_DEVICE_FACTORY(device_type, device_factory, ...) \ argument
128 #define INTERNAL_REGISTER_LOCAL_DEVICE_FACTORY(device_type, device_factory, \ argument
|
| D | device_factory.cc | 54 int32 DeviceFactory::DevicePriority(const string& device_type) { in DevicePriority()
66 void DeviceFactory::Register(const string& device_type, DeviceFactory* factory, in Register()
84 DeviceFactory* DeviceFactory::GetFactory(const string& device_type) { in GetFactory()
|
| D | memory_types.cc | 49 const DeviceType& device_type, const Graph* g, in ProcessMemoryTypes()
94 Status ValidateMemoryTypes(const DeviceType& device_type, const Graph* g) { in ValidateMemoryTypes()
151 Status EnsureMemoryTypes(const DeviceType& device_type, in EnsureMemoryTypes()
212 Status MemoryTypeForOutput(const DeviceType& device_type, const Graph* g, in MemoryTypeForOutput()
|
| D | ring_reducer_test.cc | 101 const DeviceType& device_type, in GetKernel()
113 std::unique_ptr<OpKernel> GetAdd(DataType dtype, const DeviceType& device_type, in GetAdd()
124 std::unique_ptr<OpKernel> GetDiv(DataType dtype, const DeviceType& device_type, in GetDiv()
159 const DeviceType& device_type, int num_subdivs, int fail_after) { in Init()
291 void RunTest(DataType dtype, const DeviceType& device_type, int num_workers, in RunTest()
372 const DeviceType& device_type, in GetCollectiveReduce()
411 const DeviceType& device_type, RingReducerTest* parent) in DeviceInstance()
|
| D | hierarchical_tree_broadcaster_test.cc | 216 const DeviceType& device_type, int fail_after) { in Init()
365 const DeviceType& device_type, in GetKernel()
379 const DeviceType& device_type, DeviceBase* device) { in GetCollectiveBcastSend()
397 const DeviceType& device_type, DeviceBase* device) { in GetCollectiveBcastRecv()
413 void RunTest(DataType dtype, const DeviceType& device_type, int num_workers, in RunTest()
562 const DeviceType& device_type, in DeviceInstance()
|
| D | ring_gatherer_test.cc | 101 const DeviceType& device_type, in GetKernel()
137 const DeviceType& device_type, int num_subdivs, int fail_after) { in Init()
268 void RunTest(DataType dtype, const DeviceType& device_type, int num_workers, in RunTest()
351 const DeviceType& device_type, in GetCollectiveGather()
386 const DeviceType& device_type, RingGathererTest* parent) in DeviceInstance()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.h | 40 explicit XlaPlatformInfo(const DeviceType device_type, in XlaPlatformInfo()
60 DeviceType device_type() const { return device_type_; } in device_type() function
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | constant_op_test.cc | 55 DeviceType device_type = DEVICE_CPU; in PersistentMemoryTrackingTest() local
58 DeviceType device_type = DEVICE_GPU; in PersistentMemoryTrackingTest() local
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | device_resolver_distributed_test.cc | 148 const string& device_type, in DefineWorkers()
157 void DefineWorker(const string& worker_name, const string& device_type, in DefineWorker()
181 void RestartWorker(const string& worker_name, const string& device_type, in RestartWorker()
197 const string& device_type, in ResolveIncarnationsAndValidate()
277 const string device_type = "CPU"; in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | partially_decluster_pass.cc | 70 DeviceType device_type(""); in FindNodesToDecluster() local
239 bool IsMustCompileDevice(const DeviceType& device_type) { in IsMustCompileDevice()
250 DeviceType device_type(""); in MustCompileNode() local
|
| D | device_util.cc | 62 DeviceType* device_type = id_to_device_type_.back().get(); in GetIdFor() local
91 Status DeviceNameToDeviceType(const string& device, DeviceType* device_type) { in DeviceNameToDeviceType()
|
| /external/tensorflow/tensorflow/core/profiler/internal/ |
| D | profiler_interface.h | 37 DeviceType device_type = DeviceType::kUnspecified; member
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | kernel_def.proto | 16 string device_type = 2; field
|
| D | device_attributes.proto | 37 string device_type = 2; field
|
| D | op_kernel.cc | 212 DeviceType device_type, DeviceBase* device, Allocator* allocator, in OpKernelConstruction()
1233 static string Key(StringPiece op_type, const DeviceType& device_type, in Key()
1280 const DeviceType& device_type, StringPiece node_name, in FindKernelRegistration()
1358 Status FindKernelRegistration(const DeviceType& device_type, in FindKernelRegistration()
1370 bool KernelDefAvailable(const DeviceType& device_type, in KernelDefAvailable()
1381 const DeviceType& device_type, StringPiece node_name, in FindKernelDef()
1412 Status FindKernelDef(const DeviceType& device_type, const NodeDef& node_def, in FindKernelDef()
1432 for (const DeviceType& device_type : prioritized_types) { in SupportedDeviceTypesForNode() local
1463 for (const DeviceType& device_type : prioritized_types) { in SupportedDeviceTypesForNode() local
1477 for (const DeviceType& device_type : prioritized_types) { in SupportedDeviceTypesForNode() local
[all …]
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/experimental/ |
| D | device_compatibility_check_test.py | 30 def _get_device_attrs(device_type, device_name=None, cc_major=None, argument
|
| /external/u-boot/drivers/cpu/ |
| D | riscv_cpu.c | 47 const char *device_type; in riscv_cpu_get_count() local
|
| /external/u-boot/arch/x86/cpu/ |
| D | cpu_x86.c | 67 const char *device_type; in cpu_x86_get_count() local
|
| /external/autotest/server/site_tests/bluetooth_AdapterPowerMeasure/ |
| D | bluetooth_AdapterPowerMeasure.py | 130 def run_once(self, host, max_power_mw=3, device_type='BLUETOOTH_BASE', argument
|
| /external/autotest/server/site_tests/bluetooth_AdapterSRSanity/ |
| D | bluetooth_AdapterSRSanity.py | 74 def _device_connect_async(self, device_type, device, adapter_address): argument
275 def run_peer_wakeup_device(self, device_type, device, device_test=None): argument
|
| /external/OpenCL-CTS/test_common/gl/ |
| D | setup_x11.cpp | 71 virtual int SupportsCLGLInterop( cl_device_type device_type ) in SupportsCLGLInterop()
|
| D | setup_osx.cpp | 104 virtual int SupportsCLGLInterop( cl_device_type device_type ) in SupportsCLGLInterop()
|
| /external/mesa3d/src/gallium/frontends/clover/api/ |
| D | device.cpp | 31 clGetDeviceIDs(cl_platform_id d_platform, cl_device_type device_type, in clGetDeviceIDs()
|