added LargeMemory layer

parent 30a2400e
......@@ -184,6 +184,18 @@ public class ArchitectureElementData {
return getLayerSymbol().getStringValue(AllPredefinedLayers.POOL_TYPE_NAME).get();
}
public int getNumSubKeys(){
return getLayerSymbol().getIntValue(AllPredefinedLayers.NUM_SUB_KEYS_NAME).get();
}
public int getQuerrySize(){
return getLayerSymbol().getIntValue(AllPredefinedLayers.QUERRY_SIZE_NAME).get();
}
public int getK(){
return getLayerSymbol().getIntValue(AllPredefinedLayers.K_NAME).get();
}
@Nullable
public List<Integer> getPadding(){
......
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