testEqDictStrKeyIntValue() — pytorch Function Reference

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

Function java CoreTensor Boxing calls 1

Entity Profile

CoreTensor→ Boxing→ testEqDictStrKeyIntValue() — pytorch Function Reference

Dependency Diagram

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

Relationship Graph

Source Code

android/pytorch_android/src/androidTest/java/org/pytorch/PytorchTestBase.java lines 129–151

  @Test
  public void testEqDictStrKeyIntValue() throws IOException {
    final Module module = loadModel(TEST_MODULE_ASSET_NAME);
    final Map<String, IValue> inputMap = new HashMap<>();

    inputMap.put("long_min_value", IValue.from(Long.MIN_VALUE));
    inputMap.put("long_max_value", IValue.from(Long.MAX_VALUE));
    inputMap.put("long_0", IValue.from(0l));
    inputMap.put("long_1", IValue.from(1l));
    inputMap.put("long_-1", IValue.from(-1l));

    final IValue input = IValue.dictStringKeyFrom(inputMap);
    assertTrue(input.isDictStringKey());

    final IValue output = module.runMethod("eqDictStrKeyIntValue", input);
    assertTrue(output.isDictStringKey());

    final Map<String, IValue> outputMap = output.toDictStringKey();
    assertTrue(inputMap.size() == outputMap.size());
    for (Map.Entry<String, IValue> entry : inputMap.entrySet()) {
      assertTrue(outputMap.get(entry.getKey()).toLong() == entry.getValue().toLong());
    }
  }