Class ScaleFusion

Inheritance Relationships

Base Type

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.