GetInferenceContext

Applicability

Product	Supported
Atlas 350 Accelerator Card	√
Atlas A3 training product/Atlas A3 inference product	√
Atlas A2 training product/Atlas A2 inference product	√
Atlas 200I/500 A2 inference product	√
Atlas inference product	√
Atlas training product	√

Header File/Library File

Header file: #include <graph/operator.h>
Library file: libgraph.so

Function Usage

Obtains the inference context required for the current operator's InferShape passing, for example, the shape and data type of the upstream operator.

Prototype

InferenceContextPtr GetInferenceContext() const

Parameters

None

Returns

Inference context of the current operator.

InferenceContextPtr is the alias of the pointer to class InferenceContext.

using InferenceContextPtr = std::shared_ptr<InferenceContext>;

Exception Handling

None

Restrictions

None

Parent topic: Operator