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
330 lines
10 KiB
JavaScript
330 lines
10 KiB
JavaScript
|
|
export const espdl = {};
|
|
|
|
espdl.Version = {
|
|
_START_VERSION: 0, '0': '_START_VERSION',
|
|
IR_VERSION_2023_12_22: 1, '1': 'IR_VERSION_2023_12_22'
|
|
};
|
|
|
|
espdl.AttributeType = {
|
|
UNDEFINED: 0, '0': 'UNDEFINED',
|
|
FLOAT: 1, '1': 'FLOAT',
|
|
INT: 2, '2': 'INT',
|
|
STRING: 3, '3': 'STRING',
|
|
TENSOR: 4, '4': 'TENSOR',
|
|
GRAPH: 5, '5': 'GRAPH',
|
|
FLOATS: 6, '6': 'FLOATS',
|
|
INTS: 7, '7': 'INTS',
|
|
STRINGS: 8, '8': 'STRINGS',
|
|
TENSORS: 9, '9': 'TENSORS',
|
|
GRAPHS: 10, '10': 'GRAPHS',
|
|
TYPE_FBS: 11, '11': 'TYPE_FBS',
|
|
TYPE_FBSS: 12, '12': 'TYPE_FBSS'
|
|
};
|
|
|
|
espdl.TensorDataType = {
|
|
UNDEFINED: 0, '0': 'UNDEFINED',
|
|
FLOAT: 1, '1': 'FLOAT',
|
|
UINT8: 2, '2': 'UINT8',
|
|
INT8: 3, '3': 'INT8',
|
|
UINT16: 4, '4': 'UINT16',
|
|
INT16: 5, '5': 'INT16',
|
|
INT32: 6, '6': 'INT32',
|
|
INT64: 7, '7': 'INT64',
|
|
STRING: 8, '8': 'STRING',
|
|
BOOL: 9, '9': 'BOOL',
|
|
FLOAT16: 10, '10': 'FLOAT16',
|
|
DOUBLE: 11, '11': 'DOUBLE',
|
|
UINT32: 12, '12': 'UINT32',
|
|
UINT64: 13, '13': 'UINT64'
|
|
};
|
|
|
|
espdl.DataLocation = {
|
|
DEFAULT: 0, '0': 'DEFAULT',
|
|
EXTERNAL: 1, '1': 'EXTERNAL'
|
|
};
|
|
|
|
espdl.AttributeF = class AttributeF {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new espdl.AttributeF();
|
|
$.f = reader.float32(position + 0);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
espdl.AttributeI = class AttributeI {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new espdl.AttributeI();
|
|
$.i = reader.int64(position + 0);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
espdl.Attribute = class Attribute {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new espdl.Attribute();
|
|
$.name = reader.string_(position, 4, null);
|
|
$.ref_attr_name = reader.string_(position, 6, null);
|
|
$.doc_string = reader.string_(position, 8, null);
|
|
$.attr_type = reader.int32_(position, 10, 0);
|
|
$.f = reader.struct(position, 12, espdl.AttributeF);
|
|
$.i = reader.struct(position, 14, espdl.AttributeI);
|
|
$.s = reader.array(position, 16, Uint8Array);
|
|
$.t = reader.table(position, 18, espdl.Tensor);
|
|
$.g = reader.table(position, 20, espdl.Graph);
|
|
$.tp = reader.table(position, 22, espdl.TypeInfo);
|
|
$.floats = reader.array(position, 24, Float32Array);
|
|
$.ints = reader.int64s_(position, 26);
|
|
$.strings = reader.strings_(position, 28);
|
|
$.tensors = reader.tables(position, 30, espdl.Tensor);
|
|
$.graphs = reader.tables(position, 32, espdl.Graph);
|
|
$.type_protos = reader.tables(position, 34, espdl.TypeInfo);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
espdl.ValueInfo = class ValueInfo {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new espdl.ValueInfo();
|
|
$.name = reader.string_(position, 4, null);
|
|
$.value_info_type = reader.table(position, 6, espdl.TypeInfo);
|
|
$.doc_string = reader.string_(position, 8, null);
|
|
$.exponents = reader.int64s_(position, 10);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
espdl.Node = class Node {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new espdl.Node();
|
|
$.input = reader.strings_(position, 4);
|
|
$.output = reader.strings_(position, 6);
|
|
$.name = reader.string_(position, 8, null);
|
|
$.op_type = reader.string_(position, 10, null);
|
|
$.domain = reader.string_(position, 12, null);
|
|
$.attribute = reader.tables(position, 14, espdl.Attribute);
|
|
$.doc_string = reader.string_(position, 16, null);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
espdl.Model = class Model {
|
|
|
|
static create(reader) {
|
|
return espdl.Model.decode(reader, reader.root);
|
|
}
|
|
|
|
static decode(reader, position) {
|
|
const $ = new espdl.Model();
|
|
$.ir_version = reader.int32_(position, 4, 0);
|
|
$.opset_import = reader.tables(position, 6, espdl.OperatorSetId);
|
|
$.producer_name = reader.string_(position, 8, null);
|
|
$.producer_version = reader.string_(position, 10, null);
|
|
$.domain = reader.string_(position, 12, null);
|
|
$.model_version = reader.int64_(position, 14, 0n);
|
|
$.doc_string = reader.string_(position, 16, null);
|
|
$.graph = reader.table(position, 18, espdl.Graph);
|
|
$.metadata_props = reader.tables(position, 20, espdl.StringStringEntry);
|
|
$.functions = reader.tables(position, 22, espdl.Function);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
espdl.StringStringEntry = class StringStringEntry {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new espdl.StringStringEntry();
|
|
$.key = reader.string_(position, 4, null);
|
|
$.value = reader.string_(position, 6, null);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
espdl.TensorAnnotation = class TensorAnnotation {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new espdl.TensorAnnotation();
|
|
$.tensor_name = reader.string_(position, 4, null);
|
|
$.quant_parameter_tensor_names = reader.tables(position, 6, espdl.StringStringEntry);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
espdl.Graph = class Graph {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new espdl.Graph();
|
|
$.node = reader.tables(position, 4, espdl.Node);
|
|
$.name = reader.string_(position, 6, null);
|
|
$.initializer = reader.tables(position, 8, espdl.Tensor);
|
|
$.doc_string = reader.string_(position, 10, null);
|
|
$.input = reader.tables(position, 12, espdl.ValueInfo);
|
|
$.output = reader.tables(position, 14, espdl.ValueInfo);
|
|
$.value_info = reader.tables(position, 16, espdl.ValueInfo);
|
|
$.quantization_annotation = reader.tables(position, 18, espdl.TensorAnnotation);
|
|
$.test_inputs_value = reader.tables(position, 20, espdl.Tensor);
|
|
$.test_outputs_value = reader.tables(position, 22, espdl.Tensor);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
espdl.AlignedBytes = class AlignedBytes {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new espdl.AlignedBytes();
|
|
$.bytes = reader.read(position + 0, 16);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
espdl.Tensor = class Tensor {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new espdl.Tensor();
|
|
$.dims = reader.int64s_(position, 4);
|
|
$.data_type = reader.int32_(position, 6, 0);
|
|
$.float_data = reader.array(position, 8, Float32Array);
|
|
$.int32_data = reader.array(position, 10, Int32Array);
|
|
$.string_data = reader.strings_(position, 12);
|
|
$.int64_data = reader.int64s_(position, 14);
|
|
$.name = reader.string_(position, 16, null);
|
|
$.doc_string = reader.string_(position, 18, null);
|
|
$.raw_data = reader.structs(position, 20, espdl.AlignedBytes, 16);
|
|
$.external_data = reader.tables(position, 22, espdl.StringStringEntry);
|
|
$.data_location = reader.int32_(position, 24, 0);
|
|
$.double_data = reader.array(position, 26, Float64Array);
|
|
$.uint64_data = reader.uint64s_(position, 28);
|
|
$.exponents = reader.int64s_(position, 30);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
espdl.TensorShape = class TensorShape {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new espdl.TensorShape();
|
|
$.dim = reader.tables(position, 4, espdl.Dimension);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
espdl.Dimension = class Dimension {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new espdl.Dimension();
|
|
$.value = reader.table(position, 4, espdl.DimensionValue);
|
|
$.denotation = reader.string_(position, 6, null);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
espdl.DimensionValueType = {
|
|
UNKNOWN: 0, '0': 'UNKNOWN',
|
|
VALUE: 1, '1': 'VALUE',
|
|
PARAM: 2, '2': 'PARAM'
|
|
};
|
|
|
|
espdl.DimensionValue = class DimensionValue {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new espdl.DimensionValue();
|
|
$.dim_type = reader.int8_(position, 4, 0);
|
|
$.dim_value = reader.int64_(position, 6, 0n);
|
|
$.dim_param = reader.string_(position, 8, null);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
espdl.TensorTypeAndShape = class TensorTypeAndShape {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new espdl.TensorTypeAndShape();
|
|
$.elem_type = reader.int32_(position, 4, 0);
|
|
$.shape = reader.table(position, 6, espdl.TensorShape);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
espdl.SequenceType = class SequenceType {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new espdl.SequenceType();
|
|
$.elem_type = reader.table(position, 4, espdl.TypeInfo);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
espdl.MapType = class MapType {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new espdl.MapType();
|
|
$.key_type = reader.int32_(position, 4, 0);
|
|
$.value_type = reader.table(position, 6, espdl.TypeInfo);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
espdl.OptionalType = class OptionalType {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new espdl.OptionalType();
|
|
$.elem_type = reader.table(position, 4, espdl.TypeInfo);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
espdl.TypeInfoValue = class {
|
|
|
|
static decode(reader, position, type) {
|
|
switch (type) {
|
|
case 1: return espdl.TensorTypeAndShape.decode(reader, position);
|
|
case 2: return espdl.SequenceType.decode(reader, position);
|
|
case 3: return espdl.MapType.decode(reader, position);
|
|
case 4: return espdl.OptionalType.decode(reader, position);
|
|
default: return undefined;
|
|
}
|
|
}
|
|
};
|
|
|
|
espdl.TypeInfo = class TypeInfo {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new espdl.TypeInfo();
|
|
$.value = reader.union(position, 4, espdl.TypeInfoValue);
|
|
$.denotation = reader.string_(position, 8, null);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
espdl.OperatorSetId = class OperatorSetId {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new espdl.OperatorSetId();
|
|
$.domain = reader.string_(position, 4, null);
|
|
$.version = reader.int64_(position, 6, 0n);
|
|
return $;
|
|
}
|
|
};
|
|
|
|
espdl.Function = class Function {
|
|
|
|
static decode(reader, position) {
|
|
const $ = new espdl.Function();
|
|
$.name = reader.string_(position, 4, null);
|
|
$.input = reader.strings_(position, 6);
|
|
$.output = reader.strings_(position, 8);
|
|
$.attribute = reader.strings_(position, 10);
|
|
$.attribute_proto = reader.tables(position, 12, espdl.Attribute);
|
|
$.node = reader.tables(position, 14, espdl.Node);
|
|
$.doc_string = reader.string_(position, 16, null);
|
|
$.opset_import = reader.tables(position, 18, espdl.OperatorSetId);
|
|
$.domain = reader.string_(position, 20, null);
|
|
return $;
|
|
}
|
|
};
|