Searched defs:device_type (Results 1 – 1 of 1) sorted by relevance
13 static PetscErrorCode EnumToDeviceType(TorchDeviceType device_enum, torch::DeviceType *device_type)… in EnumToDeviceType()34 …ic PetscErrorCode PetscMemTypeToDeviceType(PetscMemType mem_type, torch::DeviceType *device_type) { in PetscMemTypeToDeviceType()