SetInferenceContext

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	√

Sets the inference context for inferShape passing to the current operator, for example, the shape and data type of the upstream operator.

void SetInferenceContext(const InferenceContextPtr &inference_context)

Parameter

Input/Output

Description

inference_context

Input

Inference context of the current operator.

InferenceContextPtr is the alias of the pointer to class InferenceContext.

using InferenceContextPtr = std::shared_ptr<InferenceContext>;

None

None

None

Parent topic: Operator