Class MatMulFusion

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.