load() — pytorch Function Reference

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

Function java AndroidRuntime TensorAPI calls 1

Entity Profile

AndroidRuntime→ TensorAPI→ load() — pytorch Function Reference

Dependency Diagram

graph TD
  712d71c0_2572_73fd_03e4_66d523403f93["load()"]
  2f7e02ce_8a6b_cbd1_ccc7_157b38991295["Module()"]
  712d71c0_2572_73fd_03e4_66d523403f93 -->|calls| 2f7e02ce_8a6b_cbd1_ccc7_157b38991295
  style 712d71c0_2572_73fd_03e4_66d523403f93 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));
  }