testEqInt() — pytorch Function Reference

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

Function java CoreTensor Registration calls 1

Entity Profile

CoreTensor→ Registration→ testEqInt() — pytorch Function Reference

Dependency Diagram

graph TD
  21260c04_f1ed_0ad6_72df_38ab6df851ee["testEqInt()"]
  9f728842_eeeb_dcbf_a87a_1decf4b380f8["loadModel()"]
  21260c04_f1ed_0ad6_72df_38ab6df851ee -->|calls| 9f728842_eeeb_dcbf_a87a_1decf4b380f8
  style 21260c04_f1ed_0ad6_72df_38ab6df851ee fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

android/pytorch_android/src/androidTest/java/org/pytorch/PytorchTestBase.java lines 40–51

  @Test
  public void testEqInt() throws IOException {
    final Module module = loadModel(TEST_MODULE_ASSET_NAME);
    for (long value : new long[] {Long.MIN_VALUE, -1024, -1, 0, 1, 1024, Long.MAX_VALUE}) {
      final IValue input = IValue.from(value);
      assertTrue(input.isLong());
      assertTrue(value == input.toLong());
      final IValue output = module.runMethod("eqInt", input);
      assertTrue(output.isLong());
      assertTrue(value == output.toLong());
    }
  }