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 Oct 31, 2020
1 parent 69cfeb6 commit dee687a
Showing 1 changed file with 19 additions and 1 deletion.
20 changes: 19 additions & 1 deletion source/tflite-schema.js
Original file line number Diff line number Diff line change
Expand Up @@ -343,7 +343,8 @@ $root.tflite.BuiltinOperator = {
SEGMENT_SUM: 125,
BATCH_MATMUL: 126,
PLACEHOLDER_FOR_GREATER_OP_CODES: 127,
CUMSUM: 128
CUMSUM: 128,
CALL_ONCE: 129
};

$root.tflite.BuiltinOptions = class {
Expand Down Expand Up @@ -452,6 +453,7 @@ $root.tflite.BuiltinOptions = class {
case 100: return $root.tflite.SegmentSumOptions.decode(reader, position);
case 101: return $root.tflite.BatchMatMulOptions.decode(reader, position);
case 102: return $root.tflite.CumsumOptions.decode(reader, position);
case 103: return $root.tflite.CallOnceOptions.decode(reader, position);
}
return undefined;
}
Expand Down Expand Up @@ -560,6 +562,7 @@ $root.tflite.BuiltinOptions = class {
case 'SegmentSumOptions': return $root.tflite.SegmentSumOptions.decodeText(reader, json);
case 'BatchMatMulOptions': return $root.tflite.BatchMatMulOptions.decodeText(reader, json);
case 'CumsumOptions': return $root.tflite.CumsumOptions.decodeText(reader, json);
case 'CallOnceOptions': return $root.tflite.CallOnceOptions.decodeText(reader, json);
}
return undefined;
}
Expand Down Expand Up @@ -2051,6 +2054,21 @@ $root.tflite.IfOptions = class IfOptions {
}
};

$root.tflite.CallOnceOptions = class CallOnceOptions {

static decode(reader, position) {
const $ = new $root.tflite.CallOnceOptions();
$.init_subgraph_index = reader.int32_(position, 4, 0);
return $;
}

static decodeText(reader, json) {
const $ = new $root.tflite.CallOnceOptions();
$.init_subgraph_index = reader.value(json.init_subgraph_index, 0);
return $;
}
};

$root.tflite.WhileOptions = class WhileOptions {

static decode(reader, position) {
Expand Down

0 comments on commit dee687a

Please sign in to comment.