smokeTest() — pytorch Function Reference

Architecture documentation for the smokeTest() function in TorchVisionInstrumentedTests.java from the pytorch codebase.

Function java AndroidRuntime TensorAPI calls 1

Entity Profile

AndroidRuntime→ TensorAPI→ smokeTest() — pytorch Function Reference

Dependency Diagram

graph TD
  64738472_2c4a_2744_9ad4_b274dbe95ec3["smokeTest()"]
  7a6e88ea_1ef2_3f01_b347_e251c2d220d8["shape()"]
  64738472_2c4a_2744_9ad4_b274dbe95ec3 -->|calls| 7a6e88ea_1ef2_3f01_b347_e251c2d220d8
  style 64738472_2c4a_2744_9ad4_b274dbe95ec3 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

android/pytorch_android_torchvision/src/androidTest/java/org/pytorch/torchvision/TorchVisionInstrumentedTests.java lines 14–23

  @Test
  public void smokeTest() {
    Bitmap bitmap = Bitmap.createBitmap(320, 240, Bitmap.Config.ARGB_8888);
    Tensor tensor =
        TensorImageUtils.bitmapToFloat32Tensor(
            bitmap,
            TensorImageUtils.TORCHVISION_NORM_MEAN_RGB,
            TensorImageUtils.TORCHVISION_NORM_STD_RGB);
    assertArrayEquals(new long[] {1l, 3l, 240l, 320l}, tensor.shape());
  }