testMobileNetV2() — pytorch Function Reference

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

Function java CoreTensor Dispatcher calls 1

Entity Profile

CoreTensor→ Dispatcher→ testMobileNetV2() — pytorch Function Reference

Dependency Diagram

graph TD
  90a16d00_d8af_c7e2_1812_58613f9b4abe["testMobileNetV2()"]
  9f728842_eeeb_dcbf_a87a_1decf4b380f8["loadModel()"]
  90a16d00_d8af_c7e2_1812_58613f9b4abe -->|calls| 9f728842_eeeb_dcbf_a87a_1decf4b380f8
  style 90a16d00_d8af_c7e2_1812_58613f9b4abe 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);
    }
  }