Class PowFusion

• Defined in File pow_fusion.h

Inheritance Relationships

Base Type

• public mindspore::ops::BaseOperator

Class Documentation

class PowFusion : public mindspore::ops::BaseOperator

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.

参数

• 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.

参数

scale – [in] Define a size factor applied to input.

void set_shift(const float &shift)

Method to set shift attribute. Default is 0.0.

参数

shift – [in] Define a bias applied to input.

float get_scale() const

Method to get scale attribute.

返回

a size factor.

float get_shift() const

Method to get shift attribute.

返回

a bias value.