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