Class PReLUFusion

• Defined in File prelu_fusion.h

Inheritance Relationships

Base Type

• public mindspore::ops::BaseOperator

Class Documentation

class PReLUFusion : public mindspore::ops::BaseOperator

PReLUFusion defined PReLU operator prototype of lite.

Public Functions

inline PReLUFusion()

Constructor.

void Init(const bool channel_shared, const std::vector<float> &slope)

Method to init the op's attributes.

参数

• channel_shared – [in] Define a boolean value to indicate whether channel is shared or not.

• slope – [in] Define a size factor applied to the elements less than 0.

void set_channel_shared(const bool channel_shared)

Method to set channel_shared attribute.

参数

channel_shared – [in] Define a boolean value to indicate whether channel is shared or not.

void set_slope(const std::vector<float> &slope)

Method to set slope attribute.

参数

slope – [in] Define size factors applied to the elements less than 0.

bool get_channel_shared() const

Method to get channel_shared attribute.

返回

a boolean value.

std::vector<float> get_slope() const

Method to get slope attribute.

返回

size factors.