예제 #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
파일: init.c 프로젝트: wangg12/nxnrock
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;
}