Commit 1dd77007 authored by Christian Fuß's avatar Christian Fuß
Browse files

removed depth parameter for BeamSearch

parent 2bff4fef
Pipeline #186537 passed with stages
in 19 minutes and 42 seconds
......@@ -78,7 +78,6 @@ 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,10 +42,6 @@ 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