Exemplo n.º 1
0
 private static extern IntPtr THSTensor_max(IntPtr src, AllocatePinnedArray allocator, long dimension, bool keep_dim);
Exemplo n.º 2
0
 private static extern void THSNN_Module_get_named_parameters(HType module, AllocatePinnedArray allocator1, AllocatePinnedArray allocator2);
Exemplo n.º 3
0
 static extern void THSTensor_broadcast_tensors(IntPtr tensor, long length, AllocatePinnedArray allocator);
Exemplo n.º 4
0
 private static extern void THSNN_Optimizer_getParameters(HType module, AllocatePinnedArray allocator);
Exemplo n.º 5
0
 private static extern void THSAutograd_grad(
     IntPtr outputs, long oLength,
     IntPtr inputs, long iLength,
     IntPtr grad_outs, long gLength,
     bool retain_graph, bool create_graph, bool allow_unused,
     AllocatePinnedArray allocator);