Skip to content

Commit

Permalink
fix recurring again
Browse files Browse the repository at this point in the history
  • Loading branch information
kali committed Aug 28, 2020
1 parent b6ea1dc commit 285d260
Show file tree
Hide file tree
Showing 5 changed files with 5 additions and 1 deletion.
1 change: 1 addition & 0 deletions hir/src/ops/scan.rs
Original file line number Diff line number Diff line change
Expand Up @@ -88,6 +88,7 @@ impl InferenceScan {
input_mapping,
output_mapping,
self.seq_length_input_slot,
0
)?))
}

Expand Down
1 change: 1 addition & 0 deletions onnx/src/ops/rec/gru.rs
Original file line number Diff line number Diff line change
Expand Up @@ -346,6 +346,7 @@ impl GRU {
input_mapping,
vec![output_mapping],
self.optional_sequence_lens_input,
0,
)?,
&outer_inputs,
)?;
Expand Down
1 change: 1 addition & 0 deletions onnx/src/ops/rec/lstm.rs
Original file line number Diff line number Diff line change
Expand Up @@ -469,6 +469,7 @@ impl LSTM {
input_mapping,
vec![h_mapping, c_mapping],
self.optional_sequence_lens_input,
0,
)?,
&outer_inputs,
)?;
Expand Down
1 change: 1 addition & 0 deletions onnx/src/ops/rec/rnn.rs
Original file line number Diff line number Diff line change
Expand Up @@ -298,6 +298,7 @@ impl RNN {
input_mapping,
vec![output_mapping],
self.optional_sequence_lens_input,
0,
)?,
&outer_inputs,
)?;
Expand Down
2 changes: 1 addition & 1 deletion tensorflow/src/ops/rec/block_lstm.rs
Original file line number Diff line number Diff line change
Expand Up @@ -172,7 +172,7 @@ impl Expansion for BlockLSTM {
})
}

let scan = scan::Scan::new(body, input_mapping, output_mapping, Some(0))?;
let scan = scan::Scan::new(body, input_mapping, output_mapping, Some(0), 0)?;
model.wire_node(&*prefix, scan, &*outer_inputs)
}
}
Expand Down

0 comments on commit 285d260

Please sign in to comment.