Searched defs:device_kind (Results 1 – 6 of 6) sorted by relevance
34 std::string device_kind) in PjRtTpuDevice()
309 std::string device_kind, int node_id) in GpuDevice()
104 absl::string_view device_kind() const override { return device_kind_; } in device_kind() function
33 device_kind: DeviceKind, field
138 int device_kind) { in FakeRecordingDevice()
66 absl::string_view device_kind() const override { return device_kind_; } in device_kind() function