Searched refs:PjRtPlatformId (Results 1 – 3 of 3) sorted by relevance
49 using PjRtPlatformId = uint64; variable54 static const PjRtPlatformId kCpuId = tensorflow::Fingerprint64(kCpuName);55 static const PjRtPlatformId kGpuId = tensorflow::Fingerprint64(kGpuName);56 static const PjRtPlatformId kTpuId = tensorflow::Fingerprint64(kTpuName);171 virtual PjRtPlatformId platform_id() const = 0;
79 PjRtPlatformId platform_id() const;156 PjRtPlatformId platform_id() const override { return platform_id_; } in platform_id()248 const PjRtPlatformId platform_id_;471 PjRtPlatformId platform_id() const { return client_->platform_id(); } in platform_id()
120 PjRtPlatformId PjRtStreamExecutorDevice::platform_id() const { in platform_id()