Beispiel #1
0
int luaopen_libnnbhwd(lua_State *L)
{
  lua_newtable(L);

  nxn_FloatSpatialConvolutionUnfold_init(L);
  nxn_FloatSpatialMaxPoolingBHWD_init(L);

  nxn_DoubleSpatialConvolutionUnfold_init(L);
  nxn_DoubleSpatialMaxPoolingBHWD_init(L);

  return 1;
}
Beispiel #2
0
int luaopen_libnxn(lua_State *L)
{
  lua_newtable(L);

  nxn_FloatCrossMapNormalization_init(L);
  nxn_FloatDropmap_init(L);
  nxn_FloatLogSoftMax_init(L);
  nxn_FloatSpatialConvolutionUnfold_init(L);
  nxn_FloatSpatialMaxPoolingBHWD_init(L);

  nxn_DoubleCrossMapNormalization_init(L);
  nxn_DoubleDropmap_init(L);
  nxn_DoubleLogSoftMax_init(L);
  nxn_DoubleSpatialConvolutionUnfold_init(L);
  nxn_DoubleSpatialMaxPoolingBHWD_init(L);

  return 1;
}