testMobileNetV2() — pytorch Function Reference

Architecture documentation for the testMobileNetV2() function in PytorchTestBase.java from the pytorch codebase.

Function java AndroidRuntime TensorAPI calls 1

Entity Profile

AndroidRuntime→ TensorAPI→ testMobileNetV2() — pytorch Function Reference

Dependency Diagram

graph TD
  148f9ce0_6ed5_a652_2a43_6846ac7d3aad["testMobileNetV2()"]
  fd3711c0_c2a1_0848_309d_ba2ef1be4466["loadModel()"]
  148f9ce0_6ed5_a652_2a43_6846ac7d3aad -->|calls| fd3711c0_c2a1_0848_309d_ba2ef1be4466
  style 148f9ce0_6ed5_a652_2a43_6846ac7d3aad fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

android/pytorch_android/src/androidTest/java/org/pytorch/PytorchTestBase.java lines 488–501

  @Test
  public void testMobileNetV2() throws IOException {
    try {
      final Module module = loadModel("mobilenet_v2.ptl");
      final IValue inputs = module.runMethod("get_all_bundled_inputs");
      assertTrue(inputs.isList());
      final IValue input = inputs.toList()[0];
      assertTrue(input.isTuple());
      module.forward(input.toTuple()[0]);
      assertTrue(true);
    } catch (Exception ex) {
      assertTrue("failed to run MobileNetV2 " + ex.getMessage(), false);
    }
  }