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();
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