Commit e1876299 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

add cpu test for the rnn operator.

parent e551275d
...@@ -798,7 +798,7 @@ struct onnx_parser ...@@ -798,7 +798,7 @@ struct onnx_parser
// For RNN, LSTM, and GRU operators, one of the input arguments // For RNN, LSTM, and GRU operators, one of the input arguments
// is prim::Undefined, and it is ignored by protobuf. We use a // is prim::Undefined, and it is ignored by protobuf. We use a
// hack to ignore this argument for these three operators // hack to ignore this argument for these three operators
std::string op_type = node.op_type(); const std::string op_type = node.op_type();
if((op_type == "RNN" || op_type == "LSTM" || op_type == "GRU") && if((op_type == "RNN" || op_type == "LSTM" || op_type == "GRU") &&
input.empty() == true) input.empty() == true)
{ {
......
This diff is collapsed.
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment