checkShapeAndDataCapacityConsistency() — pytorch Function Reference

Architecture documentation for the checkShapeAndDataCapacityConsistency() function in Tensor.java from the pytorch codebase.

Function java AndroidRuntime TensorAPI calls 2 called by 2

Entity Profile

AndroidRuntime→ TensorAPI→ checkShapeAndDataCapacityConsistency() — pytorch Function Reference

Dependency Diagram

graph TD
  0c327cad_9286_e6d4_167c_ed85851324f7["checkShapeAndDataCapacityConsistency()"]
  6be65676_2065_aa9b_3e47_fa15f51dd3fe["fromBlobUnsigned()"]
  6be65676_2065_aa9b_3e47_fa15f51dd3fe -->|calls| 0c327cad_9286_e6d4_167c_ed85851324f7
  fff97aed_e384_1976_ed4b_d94a2ed220dd["fromBlob()"]
  fff97aed_e384_1976_ed4b_d94a2ed220dd -->|calls| 0c327cad_9286_e6d4_167c_ed85851324f7
  40db4560_92cc_72c7_f2bb_938fd58a0403["numel()"]
  0c327cad_9286_e6d4_167c_ed85851324f7 -->|calls| 40db4560_92cc_72c7_f2bb_938fd58a0403
  71889387_1e63_1ff3_2450_e3018d61b4ee["checkArgument()"]
  0c327cad_9286_e6d4_167c_ed85851324f7 -->|calls| 71889387_1e63_1ff3_2450_e3018d61b4ee
  style 0c327cad_9286_e6d4_167c_ed85851324f7 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

android/pytorch_android/src/main/java/org/pytorch/Tensor.java lines 693–701

  private static void checkShapeAndDataCapacityConsistency(int dataCapacity, long[] shape) {
    final long numel = numel(shape);
    checkArgument(
        numel == dataCapacity,
        "Inconsistent data capacity:%d and shape number elements:%d shape:%s",
        dataCapacity,
        numel,
        Arrays.toString(shape));
  }