Skip to content

Commit

Permalink
Update kmodel.js
Browse files Browse the repository at this point in the history
  • Loading branch information
@lutzroeder
lutzroeder committed Jan 1, 2024
1 parent 998c4dd commit a4d299f
Showing 1 changed file with 13 additions and 13 deletions.
26 changes: 13 additions & 13 deletions source/kmodel.js
View file
Original file line number Diff line number Diff line change
Expand Up @@ -736,21 +736,21 @@ kmodel.Reader = class {
layer.inputs = [ reader.parameter('input') ];
layer.outputs = [ reader.parameter('output') ];
layer.inputs[0].value[0].shape = reader.runtime_shape_t();
layer.groups = layer.int32();
layer.out_channels = layer.int32();
layer.groups = reader.int32();
layer.out_channels = reader.int32();
layer.padding_h = reader.padding();
layer.padding_w = reader.padding();
layer.filter_h = layer.int32();
layer.filter_w = layer.int32();
layer.stride_h = layer.int32();
layer.stride_w = layer.int32();
layer.dilation_h = layer.int32();
layer.dilation_w = layer.int32();
layer.input_offset = layer.int32();
layer.filter_offset = layer.int32();
layer.output_mul = layer.int32();
layer.output_shift = layer.int32();
layer.output_offset = layer.int32();
layer.filter_h = reader.int32();
layer.filter_w = reader.int32();
layer.stride_h = reader.int32();
layer.stride_w = reader.int32();
layer.dilation_h = reader.int32();
layer.dilation_w = reader.int32();
layer.input_offset = reader.int32();
layer.filter_offset = reader.int32();
layer.output_mul = reader.int32();
layer.output_shift = reader.int32();
layer.output_offset = reader.int32();
const bias = reader.span('int32', [ layer.out_channels ]);
if (bias) {
layer.inputs.push({ name: 'bias', value: [ bias ] });
Expand Down

0 comments on commit a4d299f

Please sign in to comment.