Searched defs:device_kind (Results 1 – 12 of 12) sorted by relevance
36 std::string device_kind) in PjRtTpuDevice()
367 absl::string_view PjRtCApiDevice::device_kind() const { in device_kind() function in xla::PjRtCApiDevice376 absl::string_view device_kind(args.device_kind, args.device_kind_size); in device_kind() local
406 std::string device_kind, std::string device_vendor, in GpuDevice()
111 absl::string_view device_kind() const override { return device_kind_; } in device_kind() function
114 absl::string_view TfrtCpuDevice::device_kind() const { in device_kind() function in xla::TfrtCpuDevice
49 device_kind: DeviceKind, field72 let device_kind = DeviceKind::ArmVgicV3; in new() localVariable
138 int device_kind) { in FakeRecordingDevice()
42 device_kind: DeviceKind, field
215 device_kind: DeviceKind, field
77 absl::string_view device_kind() const override { return device_kind_; } in device_kind() function
115 let device_kind = event_device.kind(); in import_event_device() localVariable
456 const char* device_kind; // out member