load() — pytorch Function Reference

Architecture documentation for the load() function in Module.java from the pytorch codebase.

Function java AndroidRuntime JavaAPI calls 1

Entity Profile

AndroidRuntime→ JavaAPI→ load() — pytorch Function Reference

Dependency Diagram

graph TD
  480a663b_0a3c_09d9_536e_211687a484d1["load()"]
  0ec9fafe_4b5b_d066_69f6_27cd6d2e7e67["Module()"]
  480a663b_0a3c_09d9_536e_211687a484d1 -->|calls| 0ec9fafe_4b5b_d066_69f6_27cd6d2e7e67
  style 480a663b_0a3c_09d9_536e_211687a484d1 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

android/pytorch_android/src/main/java/org/pytorch/Module.java lines 23–29

  public static Module load(
      final String modelPath, final Map<String, String> extraFiles, final Device device) {
    if (!NativeLoader.isInitialized()) {
      NativeLoader.init(new SystemDelegate());
    }
    return new Module(new NativePeer(modelPath, extraFiles, device));
  }