Skip to content

Commit

Permalink
Update tflite-schema.js
Browse files Browse the repository at this point in the history
  • Loading branch information
lutzroeder committed Sep 6, 2024
1 parent 691afa4 commit 14b67a9
Showing 1 changed file with 17 additions and 1 deletion.
18 changes: 17 additions & 1 deletion source/tflite-schema.js
Original file line number Diff line number Diff line change
Expand Up @@ -450,7 +450,8 @@ tflite.BuiltinOperator = {
DILATE: 203,
STABLEHLO_RNG_BIT_GENERATOR: 204,
REDUCE_WINDOW: 205,
STABLEHLO_COMPOSITE: 206
STABLEHLO_COMPOSITE: 206,
STABLEHLO_SHIFT_LEFT: 207
};

tflite.BuiltinOptions = class {
Expand Down Expand Up @@ -745,6 +746,7 @@ tflite.BuiltinOptions2 = class {
case 19: return tflite.StablehloRngBitGeneratorOptions.decode(reader, position);
case 20: return tflite.ReduceWindowOptions.decode(reader, position);
case 21: return tflite.StableHLOCompositeOptions.decode(reader, position);
case 22: return tflite.StablehloShiftLeftOptions.decode(reader, position);
default: return undefined;
}
}
Expand Down Expand Up @@ -772,6 +774,7 @@ tflite.BuiltinOptions2 = class {
case 'StablehloRngBitGeneratorOptions': return tflite.StablehloRngBitGeneratorOptions.decodeText(reader, json);
case 'ReduceWindowOptions': return tflite.ReduceWindowOptions.decodeText(reader, json);
case 'StableHLOCompositeOptions': return tflite.StableHLOCompositeOptions.decodeText(reader, json);
case 'StablehloShiftLeftOptions': return tflite.StablehloShiftLeftOptions.decodeText(reader, json);
default: return undefined;
}
}
Expand Down Expand Up @@ -3241,6 +3244,19 @@ tflite.StableHLOCompositeOptions = class StableHLOCompositeOptions {
}
};

tflite.StablehloShiftLeftOptions = class StablehloShiftLeftOptions {

static decode(/* reader, position */) {
const $ = new tflite.StablehloShiftLeftOptions();
return $;
}

static decodeText(/* reader, json */) {
const $ = new tflite.StablehloShiftLeftOptions();
return $;
}
};

tflite.Operator = class Operator {

static decode(reader, position) {
Expand Down

0 comments on commit 14b67a9

Please sign in to comment.