Commit 7cc127f0 authored by lr119628's avatar lr119628
Browse files

[update] code cleanup

parent 70819612
......@@ -21,6 +21,16 @@ public class CNNArch2GluonTemplateController extends CNNArchTemplateController {
TemplateConfiguration templateConfiguration) {
super(architecture, templateConfiguration);
}
public String getDefinedOutputDimension(){
// function calculates the output shape as defined in the .emadl, used for AdaNet layer
ArchTypeSymbol types = ((IODeclarationSymbol)this.getArchitecture().getOutputs().get(0).getDeclaration()).getType();
StringBuilder stringBuilder = new StringBuilder();
stringBuilder.append("(");
types.getDimensions().forEach(elem->stringBuilder.append(elem).append(','));
stringBuilder.append(")");
return stringBuilder.toString();
}
public void include(String relativePath, String templateWithoutFileEnding, Writer writer, NetDefinitionMode netDefinitionMode){
......
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