Searched refs:kTpuName (Results 1 – 2 of 2) sorted by relevance
53 constexpr char kTpuName[] = "tpu"; variable56 static const PjRtPlatformId kTpuId = tensorflow::Fingerprint64(kTpuName);
94 : PjRtStreamExecutorClient(kTpuName, client, std::move(devices), task_id, in PjRtTpuClient()