Home
last modified time | relevance | path

Searched refs:kGpuName (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_client.h52 constexpr char kGpuName[] = "gpu"; variable
55 static const PjRtPlatformId kGpuId = tensorflow::Fingerprint64(kGpuName);
Dgpu_device.cc337 kGpuName, xla_client, std::move(devices), in GetGpuClient()