Class PReLUFusion

• Defined in File prelu_fusion.h

Inheritance Relationships

Base Type

• public mindspore::ops::PReLU (Class PReLU)

Class Documentation

class PReLUFusion : public mindspore::ops::PReLU

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.

Parameters

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

Parameters

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.

Parameters

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

bool get_channel_shared() const

Method to get channel_shared attribute.

Returns

a boolean value.

std::vector<float> get_slope() const

Method to get slope attribute.

Returns

size factors.