Commit 02f8ea4f authored by Luis Rickert's avatar Luis Rickert Committed by lr119628
Browse files

[update]: changed formatting in ArtificialArch.ftl templae

parent 2863f404
<#assign input = element.inputs[0]>
<#if mode == "ARCHITECTURE_DEFINITION">
self.${element.name} = artificial_layer_${element.element.name}(<#list element.element.arguments as arg>${arg}=${arg} </#list>)
# ${element.name}, output shape: {<#list element.element.outputTypes as type>[${tc.join(type.dimensions, ",")}]</#list>}
self.${element.name} = artificial_layer_${element.element.name}(<#list element.element.arguments as arg>${arg}=${arg}, </#list>)
#${element.name}, output shape: {<#list element.element.outputTypes as type>[${tc.join(type.dimensions, ",")}]</#list>}
<#elseif mode == "FORWARD_FUNCTION">
${element.name} = self.${element.name}(${input})
</#if>
Supports Markdown
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