Skip to content
Snippets Groups Projects
Commit f337232c authored by Sebastian Nickels's avatar Sebastian Nickels
Browse files

Fixed bug in Const layer template

parent 028ead95
No related branches found
No related tags found
1 merge request!23Added Unroll-related features and layers
Pipeline #180752 failed
<#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
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment