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
347 lines
11 KiB
JavaScript
347 lines
11 KiB
JavaScript
|
|
export const CNTK = {};
|
|
|
|
CNTK.proto = {};
|
|
|
|
CNTK.proto.NDShape = class NDShape {
|
|
|
|
constructor() {
|
|
this.shape_dim = [];
|
|
}
|
|
|
|
static decode(reader, length) {
|
|
const message = new CNTK.proto.NDShape();
|
|
const end = length === undefined ? reader.length : reader.position + length;
|
|
while (reader.position < end) {
|
|
const tag = reader.uint32();
|
|
switch (tag >>> 3) {
|
|
case 1:
|
|
message.shape_dim = reader.array(message.shape_dim, () => reader.uint64(), tag);
|
|
break;
|
|
default:
|
|
reader.skipType(tag & 7);
|
|
break;
|
|
}
|
|
}
|
|
return message;
|
|
}
|
|
};
|
|
|
|
CNTK.proto.Axis = class Axis {
|
|
|
|
static decode(reader, length) {
|
|
const message = new CNTK.proto.Axis();
|
|
const end = length === undefined ? reader.length : reader.position + length;
|
|
while (reader.position < end) {
|
|
const tag = reader.uint32();
|
|
switch (tag >>> 3) {
|
|
case 1:
|
|
message.static_axis_idx = reader.int32();
|
|
break;
|
|
case 2:
|
|
message.name = reader.string();
|
|
break;
|
|
case 3:
|
|
message.is_ordered_dynamic_axis = reader.bool();
|
|
break;
|
|
default:
|
|
reader.skipType(tag & 7);
|
|
break;
|
|
}
|
|
}
|
|
return message;
|
|
}
|
|
};
|
|
|
|
CNTK.proto.Axis.prototype.static_axis_idx = 0;
|
|
CNTK.proto.Axis.prototype.name = "";
|
|
CNTK.proto.Axis.prototype.is_ordered_dynamic_axis = false;
|
|
|
|
CNTK.proto.NDArrayView = class NDArrayView {
|
|
|
|
get values() {
|
|
CNTK.proto.NDArrayView.valuesSet = CNTK.proto.NDArrayView.valuesSet || new Set(["float_values", "double_values", "bytes_value", "sint32_values"]);
|
|
return Object.keys(this).find((key) => CNTK.proto.NDArrayView.valuesSet.has(key) && this[key] !== null);
|
|
}
|
|
|
|
static decode(reader, length) {
|
|
const message = new CNTK.proto.NDArrayView();
|
|
const end = length === undefined ? reader.length : reader.position + length;
|
|
while (reader.position < end) {
|
|
const tag = reader.uint32();
|
|
switch (tag >>> 3) {
|
|
case 1:
|
|
message.data_type = reader.int32();
|
|
break;
|
|
case 2:
|
|
message.storage_format = reader.int32();
|
|
break;
|
|
case 3:
|
|
message.shape = CNTK.proto.NDShape.decode(reader, reader.uint32());
|
|
break;
|
|
case 4:
|
|
message.float_values = CNTK.proto.NDArrayView.FloatValues.decode(reader, reader.uint32());
|
|
break;
|
|
case 5:
|
|
message.double_values = CNTK.proto.NDArrayView.DoubleValues.decode(reader, reader.uint32());
|
|
break;
|
|
case 6:
|
|
message.bytes_value = CNTK.proto.NDArrayView.BytesValue.decode(reader, reader.uint32());
|
|
break;
|
|
case 7:
|
|
message.sint32_values = CNTK.proto.NDArrayView.IntValues.decode(reader, reader.uint32());
|
|
break;
|
|
default:
|
|
reader.skipType(tag & 7);
|
|
break;
|
|
}
|
|
}
|
|
return message;
|
|
}
|
|
};
|
|
|
|
CNTK.proto.NDArrayView.prototype.data_type = 0;
|
|
CNTK.proto.NDArrayView.prototype.storage_format = 0;
|
|
CNTK.proto.NDArrayView.prototype.shape = null;
|
|
|
|
CNTK.proto.NDArrayView.DataType = {
|
|
"Unknown": 0,
|
|
"Float": 1,
|
|
"Double": 2,
|
|
"Float16": 4,
|
|
"Int8": 5,
|
|
"Int16": 6
|
|
};
|
|
|
|
CNTK.proto.NDArrayView.StorageFormat = {
|
|
"Dense": 0,
|
|
"SparseCSC": 1,
|
|
"SparseBlockCol": 2
|
|
};
|
|
|
|
CNTK.proto.NDArrayView.FloatValues = class FloatValues {
|
|
|
|
constructor() {
|
|
this.value = [];
|
|
}
|
|
|
|
static decode(reader, length) {
|
|
const message = new CNTK.proto.NDArrayView.FloatValues();
|
|
const end = length === undefined ? reader.length : reader.position + length;
|
|
while (reader.position < end) {
|
|
const tag = reader.uint32();
|
|
switch (tag >>> 3) {
|
|
case 1:
|
|
message.value = reader.floats(message.value, tag);
|
|
break;
|
|
default:
|
|
reader.skipType(tag & 7);
|
|
break;
|
|
}
|
|
}
|
|
return message;
|
|
}
|
|
};
|
|
|
|
CNTK.proto.NDArrayView.DoubleValues = class DoubleValues {
|
|
|
|
constructor() {
|
|
this.value = [];
|
|
}
|
|
|
|
static decode(reader, length) {
|
|
const message = new CNTK.proto.NDArrayView.DoubleValues();
|
|
const end = length === undefined ? reader.length : reader.position + length;
|
|
while (reader.position < end) {
|
|
const tag = reader.uint32();
|
|
switch (tag >>> 3) {
|
|
case 1:
|
|
message.value = reader.doubles(message.value, tag);
|
|
break;
|
|
default:
|
|
reader.skipType(tag & 7);
|
|
break;
|
|
}
|
|
}
|
|
return message;
|
|
}
|
|
};
|
|
|
|
CNTK.proto.NDArrayView.BytesValue = class BytesValue {
|
|
|
|
static decode(reader, length) {
|
|
const message = new CNTK.proto.NDArrayView.BytesValue();
|
|
const end = length === undefined ? reader.length : reader.position + length;
|
|
while (reader.position < end) {
|
|
const tag = reader.uint32();
|
|
switch (tag >>> 3) {
|
|
case 1:
|
|
message.value = reader.bytes();
|
|
break;
|
|
default:
|
|
reader.skipType(tag & 7);
|
|
break;
|
|
}
|
|
}
|
|
return message;
|
|
}
|
|
};
|
|
|
|
CNTK.proto.NDArrayView.BytesValue.prototype.value = new Uint8Array([]);
|
|
|
|
CNTK.proto.NDArrayView.IntValues = class IntValues {
|
|
|
|
constructor() {
|
|
this.value = [];
|
|
}
|
|
|
|
static decode(reader, length) {
|
|
const message = new CNTK.proto.NDArrayView.IntValues();
|
|
const end = length === undefined ? reader.length : reader.position + length;
|
|
while (reader.position < end) {
|
|
const tag = reader.uint32();
|
|
switch (tag >>> 3) {
|
|
case 1:
|
|
message.value = reader.array(message.value, () => reader.sint32(), tag);
|
|
break;
|
|
default:
|
|
reader.skipType(tag & 7);
|
|
break;
|
|
}
|
|
}
|
|
return message;
|
|
}
|
|
};
|
|
|
|
CNTK.proto.Vector = class Vector {
|
|
|
|
constructor() {
|
|
this.value = [];
|
|
}
|
|
|
|
static decode(reader, length) {
|
|
const message = new CNTK.proto.Vector();
|
|
const end = length === undefined ? reader.length : reader.position + length;
|
|
while (reader.position < end) {
|
|
const tag = reader.uint32();
|
|
switch (tag >>> 3) {
|
|
case 1:
|
|
message.value.push(CNTK.proto.DictionaryValue.decode(reader, reader.uint32()));
|
|
break;
|
|
default:
|
|
reader.skipType(tag & 7);
|
|
break;
|
|
}
|
|
}
|
|
return message;
|
|
}
|
|
};
|
|
|
|
CNTK.proto.Dictionary = class Dictionary {
|
|
|
|
constructor() {
|
|
this.data = {};
|
|
}
|
|
|
|
static decode(reader, length) {
|
|
const message = new CNTK.proto.Dictionary();
|
|
const end = length === undefined ? reader.length : reader.position + length;
|
|
while (reader.position < end) {
|
|
const tag = reader.uint32();
|
|
switch (tag >>> 3) {
|
|
case 1:
|
|
message.version = reader.uint64();
|
|
break;
|
|
case 2:
|
|
reader.entry(message.data, () => reader.string(), () => CNTK.proto.DictionaryValue.decode(reader, reader.uint32()));
|
|
break;
|
|
default:
|
|
reader.skipType(tag & 7);
|
|
break;
|
|
}
|
|
}
|
|
return message;
|
|
}
|
|
};
|
|
|
|
CNTK.proto.Dictionary.prototype.version = 0n;
|
|
|
|
CNTK.proto.DictionaryValue = class DictionaryValue {
|
|
|
|
get value() {
|
|
CNTK.proto.DictionaryValue.valueSet = CNTK.proto.DictionaryValue.valueSet || new Set(["bool_value", "int_value", "size_t_value", "float_value", "double_value", "string_value", "nd_shape_value", "axis_value", "vector_value", "dictionary_value", "nd_array_view_value"]);
|
|
return Object.keys(this).find((key) => CNTK.proto.DictionaryValue.valueSet.has(key) && this[key] !== null);
|
|
}
|
|
|
|
static decode(reader, length) {
|
|
const message = new CNTK.proto.DictionaryValue();
|
|
const end = length === undefined ? reader.length : reader.position + length;
|
|
while (reader.position < end) {
|
|
const tag = reader.uint32();
|
|
switch (tag >>> 3) {
|
|
case 1:
|
|
message.version = reader.uint64();
|
|
break;
|
|
case 2:
|
|
message.value_type = reader.int32();
|
|
break;
|
|
case 3:
|
|
message.bool_value = reader.bool();
|
|
break;
|
|
case 4:
|
|
message.int_value = reader.int32();
|
|
break;
|
|
case 5:
|
|
message.size_t_value = reader.uint64();
|
|
break;
|
|
case 6:
|
|
message.float_value = reader.float();
|
|
break;
|
|
case 7:
|
|
message.double_value = reader.double();
|
|
break;
|
|
case 8:
|
|
message.string_value = reader.string();
|
|
break;
|
|
case 9:
|
|
message.nd_shape_value = CNTK.proto.NDShape.decode(reader, reader.uint32());
|
|
break;
|
|
case 10:
|
|
message.axis_value = CNTK.proto.Axis.decode(reader, reader.uint32());
|
|
break;
|
|
case 11:
|
|
message.vector_value = CNTK.proto.Vector.decode(reader, reader.uint32());
|
|
break;
|
|
case 12:
|
|
message.dictionary_value = CNTK.proto.Dictionary.decode(reader, reader.uint32());
|
|
break;
|
|
case 13:
|
|
message.nd_array_view_value = CNTK.proto.NDArrayView.decode(reader, reader.uint32());
|
|
break;
|
|
default:
|
|
reader.skipType(tag & 7);
|
|
break;
|
|
}
|
|
}
|
|
return message;
|
|
}
|
|
};
|
|
|
|
CNTK.proto.DictionaryValue.prototype.version = 0n;
|
|
CNTK.proto.DictionaryValue.prototype.value_type = 0;
|
|
|
|
CNTK.proto.DictionaryValue.Type = {
|
|
"None": 0,
|
|
"Bool": 1,
|
|
"Int": 2,
|
|
"SizeT": 3,
|
|
"Float": 4,
|
|
"Double": 5,
|
|
"String": 6,
|
|
"NDShape": 7,
|
|
"Axis": 8,
|
|
"Vector": 9,
|
|
"Dictionary": 10,
|
|
"NDArrayView": 11
|
|
};
|