Class SubFusion

Defined in File sub_fusion.h

Inheritance Relationships

Base Type

public mindspore::ops::Sub (Class Sub)

Class Documentation

class SubFusion : public mindspore::ops::Sub 

SubFusion defined Sub operator prototype of lite.

Public Functions

inline SubFusion(): Constructor.

void Init(const ActivationType &activation_type = NO_ACTIVATION)

Method to init the op’s attributes.

Parameters: activation_type – [in] Define the activation type.

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.