Home
last modified time | relevance | path

Searched refs:device_vendor (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/core/profiler/utils/
Dhardware_type_utils.cc32 if (device_cap.device_vendor() == kDeviceVendorNvidia) { in GetFmaMaxThroughputPerSMPerCycle()
81 } else if (device_cap.device_vendor() == kDeviceVendorAMD) { in GetFmaMaxThroughputPerSMPerCycle()
93 LOG(ERROR) << "Unknown device vendor " << device_cap.device_vendor(); in GetFmaMaxThroughputPerSMPerCycle()
107 if (device_cap.device_vendor() == kDeviceVendorNvidia) { in GpuModelName()
128 } else if (device_cap.device_vendor() == kDeviceVendorAMD) { in GpuModelName()
140 LOG(ERROR) << "Unknown device vendor " << device_cap.device_vendor(); in GpuModelName()
Ddevice_caps_utils.cc52 caps.device_vendor()); in SetDeviceCaps()
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dgpu_device.h35 std::string device_kind, std::string device_vendor, int node_id);
37 absl::string_view device_vendor();
Dgpu_device.cc330 description.name(), description.device_vendor(), in BuildLocalDevices()
353 device_proto->set_vendor(desc->device_vendor()); in BuildDistributedDevices()
406 std::string device_kind, std::string device_vendor, in GpuDevice() argument
410 device_vendor_(std::move(device_vendor)) { in GpuDevice()
418 absl::string_view GpuDevice::device_vendor() { return device_vendor_; } in device_vendor() function in xla::GpuDevice
/external/rust/crates/virtio-drivers/src/transport/pci/
Dbus.rs483 let device_vendor = self.root.config_read_word(current, 0); in next() localVariable
492 if device_vendor != INVALID_READ { in next()
494 let device_id = (device_vendor >> 16) as u16; in next()
495 let vendor_id = device_vendor as u16; in next()
/external/tensorflow/tensorflow/core/profiler/protobuf/
Dhardware_types.proto30 string device_vendor = 6; field
/external/rust/crates/virtio-drivers/src/transport/
Dpci.rs107 let device_vendor = root.config_read_word(device_function, 0); in new() localVariable
108 let device_id = (device_vendor >> 16) as u16; in new()
109 let vendor_id = device_vendor as u16; in new()
/external/tensorflow/tensorflow/c/experimental/stream_executor/
Dstream_executor_test.cc762 params->device->device_vendor = vendor; in TEST_F()
775 ASSERT_EQ(description.device_vendor(), "TestVendor"); in TEST_F()
789 params->device->device_vendor = vendor; in TEST_F()
801 ASSERT_EQ(description.device_vendor(), "TestVendor"); in TEST_F()
Dstream_executor.h171 const char* device_vendor; member
Dstream_executor.cc598 if (device_.device_vendor != nullptr) { in CreateDeviceDescription()
599 builder.set_device_vendor(device_.device_vendor); in CreateDeviceDescription()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Ddevice_description.cc60 result["Device Vendor"] = device_vendor(); in ToMap()
Ddevice_description.h266 const std::string &device_vendor() const { return device_vendor_; } in device_vendor() function
/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
Dtpu_executor.cc378 CHECK_NE(description->device_vendor, nullptr); in CreateDeviceDescription()
379 builder.set_device_vendor(description->device_vendor); in CreateDeviceDescription()
Dc_api_decl.h118 char* device_vendor; member
/external/pigweed/third_party/tinyusb/
DBUILD.gn34 "device_vendor",
231 pw_source_set("device_vendor") {
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorDeviceSycl.h53 device_vendor( in SyclDeviceInfo()
65 std::string device_vendor; member
689 return m_device_info.device_vendor; in getDeviceVendor()