Class PReLUFusion

Inheritance Relationships

Base Type

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.