Commit bbe7e4a4 authored by Evgeny Kusmenko's avatar Evgeny Kusmenko

Merge branch 'reshape' into 'master'

getShape

See merge request !6
parents 9f246d97 0887b261
Pipeline #201625 passed with stages
in 5 minutes and 50 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