CaptureTensor
Applicable Products
All processors
Header File/Library File
- Header file: #include <ge/fusion/pattern.h>
- Library file: libge_compiler.so
Function Usage
Captures a tensor within a pattern, allowing it to be retrieved in order from the MatchResult.
It captures the output of a node, enabling fast retrieval of the node's output via the index assigned during capture in subsequent matching. You can also retrieve the tensor description from the match results for verification.
Prototype
1 | Pattern &CaptureTensor(const NodeIo &node_output) |
Parameters
Parameter |
Input/Output |
Description |
|---|---|---|
node_output |
Input |
Tensor source, that is, an output of a node. As node_output can uniquely identify a tensor in a graph, it is often used to represent a tensor. |
Returns
Parameter |
Type |
Description |
|---|---|---|
- |
Pattern & |
Reference to the pattern for cascading calls. |
Restrictions
None
Parent topic: Pattern