setNumThreads() — pytorch Function Reference

Architecture documentation for the setNumThreads() function in PyTorchAndroid.java from the pytorch codebase.

Function java AndroidRuntime JNILayer calls 1

Entity Profile

AndroidRuntime→ JNILayer→ setNumThreads() — pytorch Function Reference

Dependency Diagram

graph TD
  b5008b24_bcaa_f2d7_4c5b_74b42adb1d4c["setNumThreads()"]
  ada484dc_76b6_1e7d_3abe_e6d4ccc1c41e["nativeSetNumThreads()"]
  b5008b24_bcaa_f2d7_4c5b_74b42adb1d4c -->|calls| ada484dc_76b6_1e7d_3abe_e6d4ccc1c41e
  style b5008b24_bcaa_f2d7_4c5b_74b42adb1d4c fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

android/pytorch_android/src/main/java/org/pytorch/PyTorchAndroid.java lines 40–46

  public static void setNumThreads(int numThreads) {
    if (numThreads < 1) {
      throw new IllegalArgumentException("Number of threads cannot be less than 1");
    }

    nativeSetNumThreads(numThreads);
  }