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