Class ScaleFusion
Defined in File scale_fusion.h
Inheritance Relationships
Base Type
public mindspore::ops::Scale
(Class Scale)
Class Documentation
-
class ScaleFusion : public mindspore::ops::Scale
ScaleFusion defined Scale operator prototype of lite.
Public Functions
-
inline ScaleFusion()
Constructor.
-
void Init(const int64_t axis = -1, const ActivationType &activation_type = NO_ACTIVATION)
Method to init the op’s attributes.
- Parameters
axis – [in] Define the first axis to do this operation.
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.
-
inline ScaleFusion()