Searched defs:device_kind (Results 1 – 7 of 7) sorted by relevance
34 std::string device_kind) in PjRtTpuDevice()
107 absl::string_view device_kind() const override { return device_kind_; } in device_kind() function
475 std::string device_kind, std::string device_vendor, in GpuDevice()
75 absl::string_view TfrtCpuDevice::device_kind() const { in device_kind() function in xla::TfrtCpuDevice
138 int device_kind) { in FakeRecordingDevice()
33 device_kind: DeviceKind, field
70 absl::string_view device_kind() const override { return device_kind_; } in device_kind() function