Device Type — pytorch Architecture

Source Code

android/pytorch_android/src/main/java/org/pytorch/Device.java lines 3–15

public enum Device {
  // Must be in sync with kDeviceCPU, kDeviceVulkan in
  // pytorch_android/src/main/cpp/pytorch_jni_lite.cpp
  CPU(1),
  VULKAN(2),
  ;

  final int jniCode;

  Device(int jniCode) {
    this.jniCode = jniCode;
  }
}