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.