Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
MatrixMultiplier Struct Reference

Public Member Functions

 MatrixMultiplier (int A, int B)
 
torch::Tensor forward (torch::Tensor weights)
 
torch::Tensor get () const
 

Detailed Description

Definition at line 7 of file extension.cpp.


The documentation for this struct was generated from the following file: