setNumThreads() — pytorch Function Reference

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

Function java AndroidRuntime TensorAPI calls 1

Entity Profile

AndroidRuntime→ TensorAPI→ setNumThreads() — pytorch Function Reference

Dependency Diagram

graph TD
  eddf10fa_0bc8_ff83_ac2d_fa502c417420["setNumThreads()"]
  eaaef05d_cae9_0a4c_971a_849011b68314["nativeSetNumThreads()"]
  eddf10fa_0bc8_ff83_ac2d_fa502c417420 -->|calls| eaaef05d_cae9_0a4c_971a_849011b68314
  style eddf10fa_0bc8_ff83_ac2d_fa502c417420 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);
  }