GetCapturedTensors

Applicability

Product

Supported or Not

Atlas A3 training products/Atlas A3 inference products

Atlas A2 training products/Atlas A2 inference products

Atlas 200I/500 A2 inference products

Atlas inference products

Atlas training products

Header File/Library File

  • Header file: #include <ge/fusion/pattern.h>
  • Library file: libge_compiler.so

Function Usage

Obtains all tensors captured within a pattern. The returned tensors are stored in a vector, ordered according to the sequence in which they were captured in the pattern definition.

Prototype

1
Status GetCapturedTensors(std::vector<NodeIo> &node_outputs) const

Parameters

Parameter

Input/Output

Description

node_outputs

Input

All tensors in a pattern.

Returns

Parameter

Type

Description

-

Status

SUCCESS: success.

FAILED: failure.

Constraints

None