Init
Description
Initializes a graph operator.
Defines the name, input, output, and shape conduction rule of a graph operator.
Definition
1 | virtual Status Init (const std::string &opName, const InferShapeFunc &inferShapeFunc, const SVector<std::string> &inTensorNames, const SVector<std::string> &outTensorNames)=0 |
Members
Member |
Description |
|---|---|
opName |
Name of a graph operator |
inferShapeFunc |
Shape conduction rule of a graph operator |
inTensorNames |
Name of the input tensor |
outTensorNames |
Name of the output tensor |
Returns
Status. If the setting is successful, NO_ERROR is returned.
Parent topic: GraphOpBuilder