GlobalPooling.ftl 234 Bytes
Newer Older
1 2 3 4 5 6
        ${element.name} = mx.symbol.Pooling(data=${element.inputs[0]},
            global_pool=True,
            kernel=(1,1),
            pool_type=${element.poolType},
            name="${element.name}")
<#include "OutputShape.ftl">