Class GpuAllocations

java.lang.Object
io.github.kirstenali.deepj.tensor.GpuAllocations

public final class GpuAllocations extends Object
Validates physical allocations without flattening logical tensor dimensions.
  • Method Details

    • requireValid

      public static void requireValid(int[] ids, int[] sizes, DType[] dtypes, int count)