Searched refs:DLDeviceTypeForDevice (Results 1 – 1 of 1) sorted by relevance
208 StatusOr<DLDeviceType> DLDeviceTypeForDevice(const PjRtDevice& device) { in DLDeviceTypeForDevice() function220 TF_ASSIGN_OR_RETURN(context.device_type, DLDeviceTypeForDevice(device)); in DLContextForDevice()