Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Ddevice_description.cc64 result["Device Vendor"] = device_vendor(); in ToMap()
Ddevice_description.h129 const std::string &device_vendor() const { return device_vendor_; } in device_vendor() function
/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_executor.cc363 CHECK_NE(description->device_vendor, nullptr); in CreateDeviceDescription()
364 builder.set_device_vendor(description->device_vendor); in CreateDeviceDescription()
Dc_api_decl.h118 char* device_vendor; member
/external/tensorflow/tensorflow/c/experimental/stream_executor/
Dstream_executor_test.cc938 params->device->device_vendor = vendor; in TEST_F()
951 ASSERT_EQ(description.device_vendor(), "TestVendor"); in TEST_F()
965 params->device->device_vendor = vendor; in TEST_F()
977 ASSERT_EQ(description.device_vendor(), "TestVendor"); in TEST_F()
Dstream_executor.h171 const char* device_vendor; member
Dstream_executor.cc613 if (device_.device_vendor != nullptr) { in CreateDeviceDescription()
614 builder.set_device_vendor(device_.device_vendor); in CreateDeviceDescription()
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dgpu_device.cc261 device_proto->set_vendor(desc->device_vendor()); in BuildDistributedDevices()