Public Member Functions | Data Fields
ngraph::op::v0::MatMul Class Reference

Operator performing Matrix Multiplication. More...

#include <matmul.hpp>

Inheritance diagram for ngraph::op::v0::MatMul:
Inheritance graph
[legend]
Collaboration diagram for ngraph::op::v0::MatMul:
Collaboration graph
[legend]

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
 

Detailed Description

Operator performing Matrix Multiplication.

Constructor & Destructor Documentation

◆ MatMul()

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.

Parameters
AMatrix A
BMatrix B
transpose_aIf matrix A should be transposed.
transpose_bIf matrix B should be transposed.

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