Class PowFusion
Defined in File pow_fusion.h
Inheritance Relationships
Base Type
public mindspore::ops::Pow
(Class Pow)
Class Documentation
-
class PowFusion : public mindspore::ops::Pow
PowFusion defined Pow operator prototype of lite.
Public Functions
-
inline PowFusion()
Constructor.
-
void Init(const float &scale, const float &shift)
Method to init the op’s attributes.
- Parameters
scale – [in] Define a size factor applied to input.
shift – [in] Define a bias applied to input.
-
void set_scale(const float &scale)
Method to set scale attribute. Default is 1.0.
- Parameters
scale – [in] Define a size factor applied to input.
-
void set_shift(const float &shift)
Method to set shift attribute. Default is 0.0.
- Parameters
shift – [in] Define a bias applied to input.
-
float get_scale() const
Method to get scale attribute.
- Returns
a size factor.
-
float get_shift() const
Method to get shift attribute.
- Returns
a bias value.
-
inline PowFusion()