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.
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.
Method to get channel_shared attribute.
- 返回
a boolean value.
-
std::vector<float> get_slope() const
Method to get slope attribute.
- 返回
size factors.
-
inline PReLUFusion()