workfunction's picture
McBopomofoLM v3.1.2 models: decoder with next-token functions (n16-n96), dec_next_first.tsv
d8edfe6 verified
Raw History Blame Contribute Delete
17.9 kB
[
{
"metadataOutputVersion" : "3.0",
"storagePrecision" : "Float16",
"outputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 3 × 15)",
"shortDescription" : "",
"shape" : "[3, 15]",
"name" : "lp",
"type" : "MultiArray"
}
],
"modelParameters" : [
],
"specificationVersion" : 9,
"functions" : [
{
"inputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Int32",
"formattedType" : "MultiArray (Int32 3 × 16)",
"shortDescription" : "",
"shape" : "[3, 16]",
"name" : "ids",
"type" : "MultiArray"
}
],
"computePrecision" : "Mixed (Float16, Float32, Int16, Int32, UInt16)",
"storagePrecision" : "Float16",
"stateSchema" : [
],
"outputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 3 × 15)",
"shortDescription" : "",
"shape" : "[3, 15]",
"name" : "lp",
"type" : "MultiArray"
}
],
"name" : "t16",
"mlProgramOperationTypeHistogram" : {
"Ios18.expandDims" : 22,
"Ios18.mul" : 250,
"Ios18.softmax" : 2,
"Ios18.rsqrt" : 39,
"Ios18.matmul" : 76,
"Ios16.reduceMean" : 39,
"Ios18.exp" : 13,
"Ios18.sigmoid" : 8,
"Ios18.greaterEqual" : 1,
"Select" : 1,
"Ios16.reduceMax" : 1,
"Tile" : 16,
"Ios18.gather" : 2,
"Ios18.add" : 123,
"Ios16.reduceSum" : 2,
"Ios18.reshape" : 68,
"Ios18.linear" : 39,
"Ios18.relu" : 6,
"Ios18.concat" : 14,
"Ios18.silu" : 20,
"Ios18.pow" : 21,
"Ios18.cast" : 5,
"Ios18.transpose" : 44,
"Ios18.abs" : 6,
"Ios18.log" : 7,
"Ios18.sub" : 2,
"Ios18.sliceByIndex" : 111
}
},
{
"inputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Int32",
"formattedType" : "MultiArray (Int32 3 × 32)",
"shortDescription" : "",
"shape" : "[3, 32]",
"name" : "ids",
"type" : "MultiArray"
}
],
"computePrecision" : "Mixed (Float16, Float32, Int16, Int32, UInt16)",
"storagePrecision" : "Float16",
"stateSchema" : [
],
"outputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 3 × 31)",
"shortDescription" : "",
"shape" : "[3, 31]",
"name" : "lp",
"type" : "MultiArray"
}
],
"name" : "t32",
"mlProgramOperationTypeHistogram" : {
"Ios18.expandDims" : 34,
"Ios18.mul" : 310,
"Ios18.softmax" : 2,
"Ios18.rsqrt" : 39,
"Ios18.matmul" : 172,
"Ios16.reduceMean" : 39,
"Ios18.exp" : 31,
"Ios18.sigmoid" : 8,
"Ios18.greaterEqual" : 1,
"Select" : 1,
"Ios16.reduceMax" : 1,
"Tile" : 16,
"Ios18.gather" : 2,
"Ios18.add" : 153,
"Ios16.reduceSum" : 2,
"Ios18.reshape" : 80,
"Ios18.linear" : 57,
"Ios18.relu" : 6,
"Ios18.concat" : 20,
"Ios18.silu" : 20,
"Ios18.pow" : 21,
"Ios18.cast" : 5,
"Ios18.transpose" : 44,
"Ios18.abs" : 6,
"Ios18.log" : 7,
"Ios18.sub" : 8,
"Ios18.sliceByIndex" : 171
}
},
{
"inputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Int32",
"formattedType" : "MultiArray (Int32 3 × 64)",
"shortDescription" : "",
"shape" : "[3, 64]",
"name" : "ids",
"type" : "MultiArray"
}
],
"computePrecision" : "Mixed (Float16, Float32, Int16, Int32, UInt16)",
"storagePrecision" : "Float16",
"stateSchema" : [
],
"outputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 3 × 63)",
"shortDescription" : "",
"shape" : "[3, 63]",
"name" : "lp",
"type" : "MultiArray"
}
],
"name" : "t64",
"mlProgramOperationTypeHistogram" : {
"Ios18.expandDims" : 70,
"Ios18.mul" : 442,
"Ios18.softmax" : 2,
"Ios18.rsqrt" : 39,
"Ios18.matmul" : 364,
"Ios16.reduceMean" : 39,
"Ios18.exp" : 79,
"Ios18.sigmoid" : 8,
"Ios18.greaterEqual" : 1,
"Select" : 1,
"Ios16.reduceMax" : 1,
"Tile" : 16,
"Ios18.gather" : 2,
"Ios18.add" : 225,
"Ios16.reduceSum" : 2,
"Ios18.reshape" : 104,
"Ios18.linear" : 105,
"Ios18.relu" : 6,
"Ios18.concat" : 20,
"Ios18.silu" : 20,
"Ios18.pow" : 21,
"Ios18.cast" : 5,
"Ios18.transpose" : 44,
"Ios18.abs" : 6,
"Ios18.log" : 7,
"Ios18.sub" : 20,
"Ios18.sliceByIndex" : 231
}
},
{
"inputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Int32",
"formattedType" : "MultiArray (Int32 3 × 96)",
"shortDescription" : "",
"shape" : "[3, 96]",
"name" : "ids",
"type" : "MultiArray"
}
],
"computePrecision" : "Mixed (Float16, Float32, Int16, Int32, UInt16)",
"storagePrecision" : "Float16",
"stateSchema" : [
],
"outputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 3 × 95)",
"shortDescription" : "",
"shape" : "[3, 95]",
"name" : "lp",
"type" : "MultiArray"
}
],
"name" : "t96",
"mlProgramOperationTypeHistogram" : {
"Ios18.expandDims" : 106,
"Ios18.mul" : 574,
"Ios18.softmax" : 2,
"Ios18.rsqrt" : 39,
"Ios18.matmul" : 556,
"Ios16.reduceMean" : 39,
"Ios18.exp" : 127,
"Ios18.sigmoid" : 8,
"Ios18.greaterEqual" : 1,
"Select" : 1,
"Ios16.reduceMax" : 1,
"Tile" : 16,
"Ios18.gather" : 2,
"Ios18.add" : 297,
"Ios16.reduceSum" : 2,
"Ios18.reshape" : 128,
"Ios18.linear" : 153,
"Ios18.relu" : 6,
"Ios18.concat" : 20,
"Ios18.silu" : 20,
"Ios18.pow" : 21,
"Ios18.cast" : 5,
"Ios18.transpose" : 44,
"Ios18.abs" : 6,
"Ios18.log" : 7,
"Ios18.sub" : 32,
"Ios18.sliceByIndex" : 291
}
},
{
"inputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Int32",
"formattedType" : "MultiArray (Int32 1 × 16)",
"shortDescription" : "",
"shape" : "[1, 16]",
"name" : "ids",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Int32",
"formattedType" : "MultiArray (Int32 1)",
"shortDescription" : "",
"shape" : "[1]",
"name" : "last",
"type" : "MultiArray"
}
],
"computePrecision" : "Mixed (Float16, Float32, Int16, Int32)",
"storagePrecision" : "Mixed (Float16, Int32)",
"stateSchema" : [
],
"outputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 1 × 16000)",
"shortDescription" : "",
"shape" : "[1, 16000]",
"name" : "lp",
"type" : "MultiArray"
}
],
"name" : "n16",
"mlProgramOperationTypeHistogram" : {
"Ios18.expandDims" : 24,
"Ios18.mul" : 250,
"Ios18.softmax" : 2,
"Ios18.rsqrt" : 39,
"Ios18.matmul" : 76,
"Ios16.reduceMean" : 39,
"Ios18.exp" : 13,
"Ios18.sigmoid" : 8,
"Ios18.equal" : 1,
"Ios18.greaterEqual" : 1,
"Select" : 1,
"Ios16.reduceMax" : 1,
"Tile" : 16,
"Ios18.gather" : 1,
"Ios18.add" : 123,
"Ios16.reduceSum" : 2,
"Ios18.reshape" : 68,
"Ios18.linear" : 39,
"Ios18.relu" : 6,
"Ios18.concat" : 14,
"Ios18.silu" : 20,
"Ios18.pow" : 21,
"Ios18.cast" : 5,
"Ios18.transpose" : 44,
"Ios18.abs" : 6,
"Ios18.log" : 7,
"Ios18.sub" : 2,
"Ios18.sliceByIndex" : 108
}
},
{
"inputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Int32",
"formattedType" : "MultiArray (Int32 1 × 32)",
"shortDescription" : "",
"shape" : "[1, 32]",
"name" : "ids",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Int32",
"formattedType" : "MultiArray (Int32 1)",
"shortDescription" : "",
"shape" : "[1]",
"name" : "last",
"type" : "MultiArray"
}
],
"computePrecision" : "Mixed (Float16, Float32, Int16, Int32)",
"storagePrecision" : "Mixed (Float16, Int32)",
"stateSchema" : [
],
"outputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 1 × 16000)",
"shortDescription" : "",
"shape" : "[1, 16000]",
"name" : "lp",
"type" : "MultiArray"
}
],
"name" : "n32",
"mlProgramOperationTypeHistogram" : {
"Ios18.expandDims" : 36,
"Ios18.mul" : 310,
"Ios18.softmax" : 2,
"Ios18.rsqrt" : 39,
"Ios18.matmul" : 172,
"Ios16.reduceMean" : 39,
"Ios18.exp" : 31,
"Ios18.sigmoid" : 8,
"Ios18.equal" : 1,
"Ios18.greaterEqual" : 1,
"Select" : 1,
"Ios16.reduceMax" : 1,
"Tile" : 16,
"Ios18.gather" : 1,
"Ios18.add" : 153,
"Ios16.reduceSum" : 2,
"Ios18.reshape" : 80,
"Ios18.linear" : 57,
"Ios18.relu" : 6,
"Ios18.concat" : 20,
"Ios18.silu" : 20,
"Ios18.pow" : 21,
"Ios18.cast" : 5,
"Ios18.transpose" : 44,
"Ios18.abs" : 6,
"Ios18.log" : 7,
"Ios18.sub" : 8,
"Ios18.sliceByIndex" : 168
}
},
{
"inputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Int32",
"formattedType" : "MultiArray (Int32 1 × 64)",
"shortDescription" : "",
"shape" : "[1, 64]",
"name" : "ids",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Int32",
"formattedType" : "MultiArray (Int32 1)",
"shortDescription" : "",
"shape" : "[1]",
"name" : "last",
"type" : "MultiArray"
}
],
"computePrecision" : "Mixed (Float16, Float32, Int16, Int32)",
"storagePrecision" : "Mixed (Float16, Int32)",
"stateSchema" : [
],
"outputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 1 × 16000)",
"shortDescription" : "",
"shape" : "[1, 16000]",
"name" : "lp",
"type" : "MultiArray"
}
],
"name" : "n64",
"mlProgramOperationTypeHistogram" : {
"Ios18.expandDims" : 72,
"Ios18.mul" : 442,
"Ios18.softmax" : 2,
"Ios18.rsqrt" : 39,
"Ios18.matmul" : 364,
"Ios16.reduceMean" : 39,
"Ios18.exp" : 79,
"Ios18.sigmoid" : 8,
"Ios18.equal" : 1,
"Ios18.greaterEqual" : 1,
"Select" : 1,
"Ios16.reduceMax" : 1,
"Tile" : 16,
"Ios18.gather" : 1,
"Ios18.add" : 225,
"Ios16.reduceSum" : 2,
"Ios18.reshape" : 104,
"Ios18.linear" : 105,
"Ios18.relu" : 6,
"Ios18.concat" : 20,
"Ios18.silu" : 20,
"Ios18.pow" : 21,
"Ios18.cast" : 5,
"Ios18.transpose" : 44,
"Ios18.abs" : 6,
"Ios18.log" : 7,
"Ios18.sub" : 20,
"Ios18.sliceByIndex" : 228
}
},
{
"inputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Int32",
"formattedType" : "MultiArray (Int32 1 × 96)",
"shortDescription" : "",
"shape" : "[1, 96]",
"name" : "ids",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Int32",
"formattedType" : "MultiArray (Int32 1)",
"shortDescription" : "",
"shape" : "[1]",
"name" : "last",
"type" : "MultiArray"
}
],
"computePrecision" : "Mixed (Float16, Float32, Int16, Int32)",
"storagePrecision" : "Mixed (Float16, Int32)",
"stateSchema" : [
],
"outputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 1 × 16000)",
"shortDescription" : "",
"shape" : "[1, 16000]",
"name" : "lp",
"type" : "MultiArray"
}
],
"name" : "n96",
"mlProgramOperationTypeHistogram" : {
"Ios18.expandDims" : 108,
"Ios18.mul" : 574,
"Ios18.softmax" : 2,
"Ios18.rsqrt" : 39,
"Ios18.matmul" : 556,
"Ios16.reduceMean" : 39,
"Ios18.exp" : 127,
"Ios18.sigmoid" : 8,
"Ios18.equal" : 1,
"Ios18.greaterEqual" : 1,
"Select" : 1,
"Ios16.reduceMax" : 1,
"Tile" : 16,
"Ios18.gather" : 1,
"Ios18.add" : 297,
"Ios16.reduceSum" : 2,
"Ios18.reshape" : 128,
"Ios18.linear" : 153,
"Ios18.relu" : 6,
"Ios18.concat" : 20,
"Ios18.silu" : 20,
"Ios18.pow" : 21,
"Ios18.cast" : 5,
"Ios18.transpose" : 44,
"Ios18.abs" : 6,
"Ios18.log" : 7,
"Ios18.sub" : 32,
"Ios18.sliceByIndex" : 288
}
}
],
"mlProgramOperationTypeHistogram" : {
"Ios18.expandDims" : 22,
"Ios18.mul" : 250,
"Ios18.softmax" : 2,
"Ios18.rsqrt" : 39,
"Ios18.matmul" : 76,
"Ios16.reduceMean" : 39,
"Ios18.exp" : 13,
"Ios18.sigmoid" : 8,
"Ios18.greaterEqual" : 1,
"Select" : 1,
"Ios16.reduceMax" : 1,
"Tile" : 16,
"Ios18.gather" : 2,
"Ios18.add" : 123,
"Ios16.reduceSum" : 2,
"Ios18.reshape" : 68,
"Ios18.linear" : 39,
"Ios18.relu" : 6,
"Ios18.concat" : 14,
"Ios18.silu" : 20,
"Ios18.pow" : 21,
"Ios18.cast" : 5,
"Ios18.transpose" : 44,
"Ios18.abs" : 6,
"Ios18.log" : 7,
"Ios18.sub" : 2,
"Ios18.sliceByIndex" : 111
},
"isUpdatable" : "0",
"stateSchema" : [
],
"availability" : {
"macOS" : "15.0",
"tvOS" : "18.0",
"visionOS" : "2.0",
"watchOS" : "11.0",
"iOS" : "18.0",
"macCatalyst" : "18.0"
},
"computePrecision" : "Mixed (Float16, Float32, Int16, Int32, UInt16)",
"modelType" : {
"name" : "MLModelType_mlProgram"
},
"inputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Int32",
"formattedType" : "MultiArray (Int32 3 × 16)",
"shortDescription" : "",
"shape" : "[3, 16]",
"name" : "ids",
"type" : "MultiArray"
}
],
"defaultFunctionName" : "t16",
"generatedClassName" : "dec60m",
"userDefinedMetadata" : {
},
"method" : "predict"
}
]