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.
-
inline PReLUFusion()