private static extern IntPtr THSNN_TransformerEncoder_forward(Module.HType module, IntPtr src, IntPtr src_mask, IntPtr src_key_padding_mask);
Esempio n. 2
0
 extern static IntPtr THSNN_Linear_forward(Module.HType module, IntPtr tensor);
Esempio n. 3
0
 extern static IntPtr THSNN_Linear_weight(Module.HType module);
Esempio n. 4
0
 extern static IntPtr THSNN_Conv1d_bias(Module.HType module);
Esempio n. 5
0
 extern static void THSNN_Conv1d_set_weight(Module.HType module, IntPtr tensor);
Esempio n. 6
0
 extern static IntPtr THSNN_ConvTranspose2d_weight(Module.HType module);
Esempio n. 7
0
 private static extern IntPtr THSNN_PixelShuffle_forward(Module.HType module, IntPtr tensor);
Esempio n. 8
0
 private static extern IntPtr THSNN_ReplicationPad1d_forward(Module.HType module, IntPtr tensor);
Esempio n. 9
0
 extern static IntPtr THSNN_RNN_forward(Module.HType module, IntPtr input, IntPtr h_0, out IntPtr h_n);
Esempio n. 10
0
 private static extern IntPtr THSNN_Sequential_forward(Module.HType module, IntPtr tensor);
Esempio n. 11
0
 private static extern IntPtr THSNN_ConstantPad2d_forward(Module.HType module, IntPtr tensor);
Esempio n. 12
0
 private static extern void THSNN_Sequential_push_back(Module.HType module,
                                                       [MarshalAs(UnmanagedType.LPStr)] string name,
                                                       BoxedModule.HType boxedSubModule);
Esempio n. 13
0
 private static extern IntPtr THSNN_ReflectionPad2d_forward(Module.HType module, IntPtr tensor);
Esempio n. 14
0
 private static extern IntPtr THSNN_TransformerEncoder_ctor(Module.HType encoder_layer, long num_layers, out IntPtr pBoxedModule);
Esempio n. 15
0
 extern static IntPtr THSNN_ConvTranspose2d_bias(Module.HType module);
Esempio n. 16
0
 private static extern IntPtr THSNN_Embedding_forward(Module.HType module, IntPtr tensor);
Esempio n. 17
0
 extern static void THSNN_ConvTranspose2d_set_bias(Module.HType module, IntPtr tensor);
Esempio n. 18
0
 extern static IntPtr THSNN_Embedding_weight(Module.HType module);
Esempio n. 19
0
 private static extern IntPtr THSNN_LogSoftmax_forward(Module.HType handle, IntPtr tensor);
Esempio n. 20
0
 extern static void THSNN_Embedding_set_weight(Module.HType module, IntPtr tensor);
Esempio n. 21
0
 private static extern IntPtr THSNN_RReLU_forward(Module.HType module, IntPtr tensor);
Esempio n. 22
0
 extern static IntPtr THSNN_Bilinear_forward(Module.HType module, IntPtr input1, IntPtr input2);
Esempio n. 23
0
 extern static IntPtr THSNN_Conv1d_weight(Module.HType module);
Esempio n. 24
0
 extern static void THSNN_Bilinear_set_bias(Module.HType module, IntPtr tensor);
Esempio n. 25
0
 private static extern IntPtr THSNN_MaxPool1d_forward(Module.HType module, IntPtr tensor);
Esempio n. 26
0
 private static extern IntPtr THSNN_Softmax2d_forward(Module.HType module, IntPtr tensor);
Esempio n. 27
0
 extern static IntPtr THSNN_Linear_bias(Module.HType module);
Esempio n. 28
0
 private static extern IntPtr THSNN_ConvTranspose2d_forward(Module.HType module, IntPtr tensor);
Esempio n. 29
0
 extern static void THSNN_Linear_set_weight(Module.HType module, IntPtr tensor);
Esempio n. 30
0
 extern static IntPtr THSNN_linearModuleApply(Module.HType module, IntPtr tensor);