Commit 0887b261 authored by Eyüp Harputlu's avatar Eyüp Harputlu

getShape

parent 140646a3
Pipeline #201488 passed with stages
in 3 minutes and 48 seconds
......@@ -161,6 +161,10 @@ public class ArchitectureElementData {
return getLayerSymbol().getBooleanValue(AllPredefinedLayers.FLATTEN_PARAMETER_NAME).get();
}
public List<Integer> getShape() {
return getLayerSymbol().getIntTupleValue(AllPredefinedLayers.SHAPE_NAME).get();
}
@Nullable
public String getPoolType(){
return getLayerSymbol().getStringValue(AllPredefinedLayers.POOL_TYPE_NAME).get();
......
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