Const.ftl 351 Bytes
Newer Older
1
<#if mode == "FORWARD_FUNCTION">
2
        ${element.name} = gluon.Constant('${element.name}', ${element.constValue})
3
<#elseif mode == "PYTHON_INLINE">
4
                    ${element.name} = mx.nd.full((batch_size, 1,), ${element.constValue}, ctx=mx_context)
5
6
7
<#elseif mode == "CPP_INLINE">
    vector<float> ${element.name}{${element.constValue}};
</#if>