Пример #1
0
int luaopen_libnxn(lua_State *L)
{
  lua_newtable(L);
  lua_pushvalue(L, -1);
  lua_setfield(L, LUA_GLOBALSINDEX, "nxn");

  nxn_FloatReLU_init(L);
  nxn_DoubleReLU_init(L);

  nxn_FloatJitter_init(L);
  nxn_DoubleJitter_init(L);

  nxn_FloatDropmap_init(L);
  nxn_DoubleDropmap_init(L);

  nxn_FloatLogSoftMax_init(L);
  nxn_DoubleLogSoftMax_init(L);

  nxn_FloatSpatialMaxPooling_init(L);
  nxn_DoubleSpatialMaxPooling_init(L);

  nxn_FloatCrossMapNormalization_init(L);
  nxn_DoubleCrossMapNormalization_init(L);

  nxn_FloatSpatialConvolution_init(L);
  nxn_DoubleSpatialConvolution_init(L);

  nxn_FloatConvProto_init(L);
  nxn_DoubleConvProto_init(L);

  return 1;
}
Пример #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;
}