Class PartialFusion
Defined in File partial_fusion.h
Inheritance Relationships
Base Type
public mindspore::ops::BaseOperator
Class Documentation
-
class PartialFusion : public mindspore::ops::BaseOperator
PartialFusion defined Partial operator prototype of lite.
Public Functions
-
inline PartialFusion()
Constructor.
-
void Init(const int64_t sub_graph_index)
Method to init the op’s attributes.
- Parameters
sub_graph_index – [in] Define a index value to indicate which sub-graph.
-
void set_sub_graph_index(const int64_t sub_graph_index)
Method to set sub_graph_index attribute.
- Parameters
sub_graph_index – [in] Define a index value to indicate which sub-graph.
-
int64_t get_sub_graph_index() const
Method to get sub_graph_index attribute.
- Returns
sub-graph index.
-
inline PartialFusion()