@@ -17,18 +17,18 @@ EXPORT_API(int) THSNN_Module_is_training(NNModule module);
17
17
EXPORT_API (void ) THSNN_Module_train (NNModule module );
18
18
EXPORT_API (void ) THSNN_Module_eval (NNModule module );
19
19
EXPORT_API (long ) THSNN_Module_children_size (const NNModule module );
20
- EXPORT_API (const char * ) THSNN_Module_child_name (const NNModule module , const int index );
20
+ EXPORT_API (NNModule ) THSNN_Module_child (const NNModule module , const int index );
21
21
EXPORT_API (const char * ) THSNN_Module_name (const NNModule module );
22
22
EXPORT_API (void ) THSNN_Module_zero_grad (const NNModule module );
23
23
EXPORT_API (void ) THSNN_Module_save (const NNModule module , const char * location );
24
24
EXPORT_API (NNModule ) THSNN_Module_load (const char * location , const char * name );
25
- EXPORT_API (NNModule ) THSNN_Module_register_module (const NNModule module , const char * name , const NNModule submodule );
25
+ EXPORT_API (void ) THSNN_Module_register_module (const NNModule module , const char * name , const NNModule submodule );
26
26
EXPORT_API (void ) THSNN_Module_dispose (const NNModule module );
27
27
28
28
EXPORT_API (void ) THSNN_AnyModule_dispose (const NNAnyModule module );
29
29
//EXPORT_API(NNModule) THSNN_AnyModule_get(const NNAnyModule module);
30
30
31
- EXPORT_API (NNModule ) THSNN_custom_module (const char * * names , at ::Tensor * * parameters , const bool * require_grad , const int length , Tensor (* forward )(Tensor ), NNAnyModule * outAsAnyModule );
31
+ EXPORT_API (NNModule ) THSNN_custom_module (const char * name , const char * * names , at ::Tensor * * parameters , const bool * require_grad , const int length , Tensor (* forward )(Tensor ), NNAnyModule * outAsAnyModule );
32
32
EXPORT_API (NNModule ) THSNN_AdaptiveAvgPool2d_ctor (const int64_t * sizes , const int length , NNAnyModule * outAsAnyModule );
33
33
EXPORT_API (Tensor ) THSNN_AdaptiveAvgPool2d_forward (const NNModule module , const Tensor tensor );
34
34
@@ -57,19 +57,19 @@ EXPORT_API(void) THSNN_Linear_set_weight(const NNModule module, const Tensor
57
57
EXPORT_API (NNModule ) THSNN_ReLU_ctor (bool inplace , NNAnyModule * outAsAnyModule );
58
58
EXPORT_API (Tensor ) THSNN_ReLU_forward (const NNModule module , const Tensor tensor );
59
59
60
- EXPORT_API (NNSequential ) THSNN_Sequential_ctor ();
61
- EXPORT_API (void ) THSNN_Sequential_push_back (const NNSequential module , const char * name , const NNAnyModule submodule );
62
- EXPORT_API (Tensor ) THSNN_Sequential_forward (const NNSequential module , const Tensor tensor );
60
+ EXPORT_API (NNModule ) THSNN_Sequential_ctor ();
61
+ EXPORT_API (void ) THSNN_Sequential_push_back (const NNModule module , const char * name , const NNAnyModule submodule );
62
+ EXPORT_API (Tensor ) THSNN_Sequential_forward (const NNModule module , const Tensor tensor );
63
63
64
64
EXPORT_API (void ) THSNN_Optimizer_zeroGrad (const Optimizer optimizer );
65
65
EXPORT_API (void ) THSNN_Optimizer_getParameters (const Optimizer optimizer , Tensor * (* allocator )(size_t length ));
66
66
EXPORT_API (void ) THSNN_Optimizer_step (const Optimizer optimizer );
67
67
EXPORT_API (void ) THSNN_Optimizer_dispose (const Optimizer optimizer );
68
68
69
- EXPORT_API (Tensor ) THSTorch_binary_cross_entropy (const Tensor inputwrapper , const Tensor targetwrapper , const Tensor weightwrapper , const int64_t reduction );
70
- EXPORT_API (Tensor ) THSTorch_mse_loss (const Tensor inputwrapper , const Tensor targetwrapper , const int64_t reduction );
71
- EXPORT_API (Tensor ) THSTorch_nll_loss (const Tensor inputwrapper , const Tensor targetwrapper , const Tensor weightwrapper , const int64_t reduction );
72
- EXPORT_API (Tensor ) THSTorch_poisson_nll_loss (const Tensor input , const Tensor target , const bool logInput , const bool full , const double eps , const int64_t reduction );
69
+ EXPORT_API (Tensor ) THSNN_binary_cross_entropy (const Tensor inputwrapper , const Tensor targetwrapper , const Tensor weightwrapper , const int64_t reduction );
70
+ EXPORT_API (Tensor ) THSNN_mse_loss (const Tensor inputwrapper , const Tensor targetwrapper , const int64_t reduction );
71
+ EXPORT_API (Tensor ) THSNN_nll_loss (const Tensor inputwrapper , const Tensor targetwrapper , const Tensor weightwrapper , const int64_t reduction );
72
+ EXPORT_API (Tensor ) THSNN_poisson_loss (const Tensor input , const Tensor target , const bool logInput , const bool full , const double eps , const int64_t reduction );
73
73
74
74
EXPORT_API (Optimizer ) THSNN_Adam_ctor (const Tensor * parameters , const int len , const double learnig_rate );
75
75
0 commit comments