Operator performing Matrix Multiplication. More...
#include <matmul.hpp>
Public Member Functions | |
MatMul (const Output< Node > &A, const Output< Node > &B, const bool &transpose_a=0, const bool &transpose_b=0) | |
Constructs an Matrix Multiplication operation. More... | |
bool | visit_attributes (AttributeVisitor &visitor) override |
virtual void | pre_validate_and_infer_types () override |
virtual OutputVector | decompose_op () const override |
virtual std::shared_ptr< Node > | clone_with_new_inputs (const OutputVector &new_args) const override |
bool | evaluate (const HostTensorVector &outputs, const HostTensorVector &inputs) const override |
bool | get_transpose_a () const |
bool | get_transpose_b () const |
Data Fields | |
NGRAPH_RTTI_DECLARATION | |
Operator performing Matrix Multiplication.
ngraph::op::v0::MatMul::MatMul | ( | const Output< Node > & | A, |
const Output< Node > & | B, | ||
const bool & | transpose_a = 0 , |
||
const bool & | transpose_b = 0 |
||
) |
Constructs an Matrix Multiplication operation.
A | Matrix A |
B | Matrix B |
transpose_a | If matrix A should be transposed. |
transpose_b | If matrix B should be transposed. |