Class MatMulFusion

• Defined in File mat_mul_fusion.h

Inheritance Relationships

Base Type

• public mindspore::ops::BaseOperator

Class Documentation

class MatMulFusion : public mindspore::ops::BaseOperator

Multiplies matrix a and matrix b. Refer to Python API mindspore.ops.MatMul for more details.

Public Functions

inline MatMulFusion()

Constructor.

void Init(bool transpose_a = false, bool transpose_b = false, const ActivationType &activation_type = NO_ACTIVATION)

Init. Refer to the parameters of Python API mindspore.ops.MatMulFusion for the inputs.

void set_activation_type(const ActivationType activation_type)

Method to set activation type.

参数

activation_type – [in] Define the activation type.

ActivationType get_activation_type() const

Method to get activation type.

返回

activation type.

void set_transpose_a(bool transpose_a)

Set transpose_a.

void set_transpose_b(bool transpose_b)

Set transpose_b.

bool get_transpose_a() const

Get transpose_a.

返回

transpose_a.

bool get_transpose_b() const

Get transpose_b.

返回

transpose_b.