Class PowFusion

Inheritance Relationships

Base Type

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.