wehub-resource-sync
7254f7b4d1
Build / Build (macos-latest) (push) Has been cancelled
Build / Build (ubuntu-latest) (push) Has been cancelled
Build / Build (windows-latest) (push) Has been cancelled
Build / Analyze (javascript) (push) Has been cancelled
Build / Analyze (python) (push) Has been cancelled
351 lines
12 KiB
JavaScript
351 lines
12 KiB
JavaScript
|
|
export const torch = {};
|
|
|
|
torch.jit = torch.jit || {};
|
|
|
|
torch.jit.mobile = torch.jit.mobile || {};
|
|
|
|
torch.jit.mobile.serialization = torch.jit.mobile.serialization || {};
|
|
|
|
torch.jit.mobile.serialization.Int = class Int {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new torch.jit.mobile.serialization.Int();
|
|
$.int_val = reader.int64(position + 0);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
torch.jit.mobile.serialization.Bool = class Bool {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new torch.jit.mobile.serialization.Bool();
|
|
$.bool_val = reader.bool(position + 0);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
torch.jit.mobile.serialization.Double = class Double {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new torch.jit.mobile.serialization.Double();
|
|
$.double_val = reader.float64(position + 0);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
torch.jit.mobile.serialization.PerTensorAffineSchema = class PerTensorAffineSchema {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new torch.jit.mobile.serialization.PerTensorAffineSchema();
|
|
$.q_scale = reader.float64(position + 0);
|
|
$.q_zero_point = reader.int32(position + 4);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
torch.jit.mobile.serialization.QuantizedSchema = class QuantizedSchema {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new torch.jit.mobile.serialization.QuantizedSchema();
|
|
$.qscheme = reader.int8_(position, 4, 0);
|
|
$.scale = reader.float64_(position, 6, 0);
|
|
$.zero_point = reader.int32_(position, 8, 0);
|
|
$.scales = reader.table(position, 10, torch.jit.mobile.serialization.TensorMetadata);
|
|
$.zero_points = reader.table(position, 12, torch.jit.mobile.serialization.TensorMetadata);
|
|
$.axis = reader.int32_(position, 14, 0);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
torch.jit.mobile.serialization.TensorMetadata = class TensorMetadata {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new torch.jit.mobile.serialization.TensorMetadata();
|
|
$.storage_location_index = reader.uint32_(position, 4, 0);
|
|
$.scalar_type = reader.int8_(position, 6, 0);
|
|
$.storage_offset = reader.int32_(position, 8, 0);
|
|
$.sizes = reader.array(position, 10, Int32Array);
|
|
$.strides = reader.array(position, 12, Int32Array);
|
|
$.requires_grad = reader.bool_(position, 14, false);
|
|
$.quantized_schema = reader.table(position, 16, torch.jit.mobile.serialization.QuantizedSchema);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
torch.jit.mobile.serialization.String = class String {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new torch.jit.mobile.serialization.String();
|
|
$.data = reader.string_(position, 4, null);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
torch.jit.mobile.serialization.Device = class Device {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new torch.jit.mobile.serialization.Device();
|
|
$.str = reader.string_(position, 4, null);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
torch.jit.mobile.serialization.List = class List {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new torch.jit.mobile.serialization.List();
|
|
$.items = reader.array(position, 4, Uint32Array);
|
|
$.annotation_str = reader.string_(position, 6, null);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
torch.jit.mobile.serialization.IntList = class IntList {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new torch.jit.mobile.serialization.IntList();
|
|
$.items = reader.int64s_(position, 4);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
torch.jit.mobile.serialization.DoubleList = class DoubleList {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new torch.jit.mobile.serialization.DoubleList();
|
|
$.items = reader.array(position, 4, Float64Array);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
torch.jit.mobile.serialization.BoolList = class BoolList {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new torch.jit.mobile.serialization.BoolList();
|
|
$.items = reader.bools_(position, 4);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
torch.jit.mobile.serialization.Tuple = class Tuple {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new torch.jit.mobile.serialization.Tuple();
|
|
$.items = reader.array(position, 4, Uint32Array);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
torch.jit.mobile.serialization.Dict = class Dict {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new torch.jit.mobile.serialization.Dict();
|
|
$.keys = reader.array(position, 4, Uint32Array);
|
|
$.values = reader.array(position, 6, Uint32Array);
|
|
$.annotation_str = reader.string_(position, 8, null);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
torch.jit.mobile.serialization.TypeType = {
|
|
UNSET: 0, '0': 'UNSET',
|
|
CLASS_WITH_FIELD: 1, '1': 'CLASS_WITH_FIELD',
|
|
CUSTOM_CLASS: 2, '2': 'CUSTOM_CLASS',
|
|
CLASS_WITH_SETSTATE: 3, '3': 'CLASS_WITH_SETSTATE',
|
|
NON_OBJ: 4, '4': 'NON_OBJ'
|
|
};
|
|
|
|
torch.jit.mobile.serialization.ObjectType = class ObjectType {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new torch.jit.mobile.serialization.ObjectType();
|
|
$.type_name = reader.string_(position, 4, null);
|
|
$.type = reader.uint8_(position, 6, 0);
|
|
$.attr_names = reader.strings_(position, 8);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
torch.jit.mobile.serialization.Object = class Object {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new torch.jit.mobile.serialization.Object();
|
|
$.type_index = reader.uint32_(position, 4, 0);
|
|
$.state = reader.uint32_(position, 6, 0);
|
|
$.attrs = reader.array(position, 8, Uint32Array);
|
|
$.setstate_func = reader.uint32_(position, 10, 0);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
torch.jit.mobile.serialization.ComplexDouble = class ComplexDouble {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new torch.jit.mobile.serialization.ComplexDouble();
|
|
$.real = reader.float64(position + 0);
|
|
$.imag = reader.float64(position + 4);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
torch.jit.mobile.serialization.EnumValue = class EnumValue {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new torch.jit.mobile.serialization.EnumValue();
|
|
$.type_name = reader.string_(position, 4, null);
|
|
$.value = reader.uint32_(position, 6, 0);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
torch.jit.mobile.serialization.Instruction = class Instruction {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new torch.jit.mobile.serialization.Instruction();
|
|
$.op = reader.int8(position + 0);
|
|
$.n = reader.uint16(position + 2);
|
|
$.x = reader.int32(position + 4);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
torch.jit.mobile.serialization.Operator = class Operator {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new torch.jit.mobile.serialization.Operator();
|
|
$.name = reader.string_(position, 4, null);
|
|
$.overload_name = reader.string_(position, 6, null);
|
|
$.num_args_serialized = reader.int32_(position, 8, -1);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
torch.jit.mobile.serialization.Arg = class Arg {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new torch.jit.mobile.serialization.Arg();
|
|
$.name = reader.string_(position, 4, null);
|
|
$.type = reader.string_(position, 6, null);
|
|
$.default_value = reader.uint32_(position, 8, 0);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
torch.jit.mobile.serialization.Schema = class Schema {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new torch.jit.mobile.serialization.Schema();
|
|
$.arguments = reader.tables(position, 4, torch.jit.mobile.serialization.Arg);
|
|
$.returns = reader.tables(position, 6, torch.jit.mobile.serialization.Arg);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
torch.jit.mobile.serialization.DebugInfo = class DebugInfo {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new torch.jit.mobile.serialization.DebugInfo();
|
|
$.debug_handle = reader.int64s_(position, 4);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
torch.jit.mobile.serialization.Function = class Function {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new torch.jit.mobile.serialization.Function();
|
|
$.qn = reader.string_(position, 4, null);
|
|
$.instructions = reader.structs(position, 6, torch.jit.mobile.serialization.Instruction, 8);
|
|
$.operators = reader.tables(position, 8, torch.jit.mobile.serialization.Operator);
|
|
$.constants = reader.array(position, 10, Uint32Array);
|
|
$.type_annotations = reader.strings_(position, 12);
|
|
$.register_size = reader.int32_(position, 14, 0);
|
|
$.schema = reader.table(position, 16, torch.jit.mobile.serialization.Schema);
|
|
$.debug_info = reader.table(position, 18, torch.jit.mobile.serialization.DebugInfo);
|
|
$.class_type = reader.uint32_(position, 20, 0);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
torch.jit.mobile.serialization.StorageData = class StorageData {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new torch.jit.mobile.serialization.StorageData();
|
|
$.data = reader.array(position, 4, Uint8Array);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
torch.jit.mobile.serialization.IValueUnion = class {
|
|
|
|
static decode(reader, position, type) {
|
|
switch (type) {
|
|
case 1: return torch.jit.mobile.serialization.Int.decode(reader, position);
|
|
case 2: return torch.jit.mobile.serialization.Bool.decode(reader, position);
|
|
case 3: return torch.jit.mobile.serialization.Double.decode(reader, position);
|
|
case 4: return torch.jit.mobile.serialization.ComplexDouble.decode(reader, position);
|
|
case 5: return torch.jit.mobile.serialization.TensorMetadata.decode(reader, position);
|
|
case 6: return torch.jit.mobile.serialization.String.decode(reader, position);
|
|
case 7: return torch.jit.mobile.serialization.List.decode(reader, position);
|
|
case 8: return torch.jit.mobile.serialization.Tuple.decode(reader, position);
|
|
case 9: return torch.jit.mobile.serialization.Dict.decode(reader, position);
|
|
case 10: return torch.jit.mobile.serialization.Object.decode(reader, position);
|
|
case 11: return torch.jit.mobile.serialization.IntList.decode(reader, position);
|
|
case 12: return torch.jit.mobile.serialization.DoubleList.decode(reader, position);
|
|
case 13: return torch.jit.mobile.serialization.BoolList.decode(reader, position);
|
|
case 14: return torch.jit.mobile.serialization.Device.decode(reader, position);
|
|
case 15: return torch.jit.mobile.serialization.EnumValue.decode(reader, position);
|
|
case 16: return torch.jit.mobile.serialization.Function.decode(reader, position);
|
|
default: return undefined;
|
|
}
|
|
}
|
|
};
|
|
|
|
torch.jit.mobile.serialization.IValue = class IValue {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new torch.jit.mobile.serialization.IValue();
|
|
$.val = reader.union(position, 4, torch.jit.mobile.serialization.IValueUnion);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
torch.jit.mobile.serialization.ExtraFile = class ExtraFile {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new torch.jit.mobile.serialization.ExtraFile();
|
|
$.name = reader.string_(position, 4, null);
|
|
$.content = reader.string_(position, 6, null);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
torch.jit.mobile.serialization.Module = class Module {
|
|
|
|
static identifier(reader) {
|
|
return reader.identifier === 'PTMF';
|
|
}
|
|
|
|
static create(reader) {
|
|
return torch.jit.mobile.serialization.Module.decode(reader, reader.root);
|
|
}
|
|
|
|
static decode(reader, position) {
|
|
const $ = new torch.jit.mobile.serialization.Module();
|
|
$.bytecode_version = reader.uint32_(position, 4, 0);
|
|
$.extra_files = reader.tables(position, 6, torch.jit.mobile.serialization.ExtraFile);
|
|
$.methods = reader.array(position, 8, Uint32Array);
|
|
$.state_obj = reader.uint32_(position, 10, 0);
|
|
$.ivalues = reader.tables(position, 12, torch.jit.mobile.serialization.IValue);
|
|
$.storage_data_size = reader.int32_(position, 14, 0);
|
|
$.storage_data = reader.tables(position, 16, torch.jit.mobile.serialization.StorageData);
|
|
$.object_types = reader.tables(position, 18, torch.jit.mobile.serialization.ObjectType);
|
|
$.jit_sources = reader.tables(position, 20, torch.jit.mobile.serialization.ExtraFile);
|
|
$.jit_constants = reader.array(position, 22, Uint32Array);
|
|
$.operator_version = reader.uint32_(position, 24, 0);
|
|
$.mobile_ivalue_size = reader.uint32_(position, 26, 0);
|
|
return $;
|
|
}
|
|
};
|