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