Class MatMulFusion

• Defined in File mat_mul_fusion.h

Inheritance Relationships

Base Type

• public MatMul

Class Documentation

class MatMulFusion : public MatMul

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.

Parameters

activation_type – [in] Define the activation type.

ActivationType get_activation_type() const

Method to get activation type.

Returns

activation type.