Skip to content

GitLab

Commit f337232c authored by Sebastian N.'s avatar Sebastian N.
Browse files

Fixed bug in Const layer template

parent 028ead95
Pipeline #180752 failed with stages
in 56 seconds
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/main/resources/templates/gluon/elements/Const.ftl
View file @ f337232c
<#if mode == "FORWARD_FUNCTION">
${element.name} = gluon.Const('${element.name}', ${element.constValue})
${element.name} = gluon.Constant('${element.name}', ${element.constValue})
<#elseif mode == "PYTHON_INLINE">
${element.name} = nd.array(${element.constValue})
${element.name} = mx.nd.full((batch_size, 1,), ${element.constValue}, ctx=mx_context)
<#elseif mode == "CPP_INLINE">
vector<float> ${element.name}{${element.constValue}};
</#if>
\ No newline at end of file
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