Commit 2bff4fef authored by Christian Fuß's avatar Christian Fuß
Browse files

added depth parameter for BeamSearch

parent 165a3d2b
Pipeline #183511 failed with stages
......@@ -78,6 +78,7 @@ public class AllPredefinedLayers {
public static final String FLATTEN_PARAMETER_NAME = "flatten";
public static final String MAX_LENGTH_NAME = "max_length";
public static final String WIDTH_NAME = "width";
public static final String DEPTH_NAME = "depth";
//possible String values
public static final String PADDING_VALID = "valid";
......
......@@ -42,6 +42,10 @@ public class BeamSearch extends UnrollDeclarationSymbol {
new ParameterSymbol.Builder()
.name(AllPredefinedLayers.WIDTH_NAME)
.constraints(Constraints.INTEGER, Constraints.POSITIVE)
.build(),
new ParameterSymbol.Builder()
.name(AllPredefinedLayers.DEPTH_NAME)
.constraints(Constraints.INTEGER, Constraints.POSITIVE)
.build()));
declaration.setParameters(parameters);
return declaration;
......
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