提交 59483f70 编写于 作者: K kechxu 提交者: Jiangtao Hu

Prediction: add Conv1d parameter proto

上级 c097b926
......@@ -21,6 +21,13 @@ message DenseParameter {
optional TensorParameter bias = 5;
}
message Conv1dParameter {
optional int32 units = 1;
optional bool use_bias = 2;
optional TensorParameter weights = 3;
optional TensorParameter bias = 4;
}
message BatchNormalizationParameter {
optional double epsilon = 1 [default = 1e-5];
optional int32 axis = 2;
......@@ -83,5 +90,6 @@ message LayerParameter {
LSTMParameter lstm = 8;
FlattenParameter flatten = 9;
ConcatenateParameter concatenate = 10;
Conv1dParameter conv1d = 11;
}
}
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册