目录

SetInferenceContext

Function Description

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

Function Prototype

void SetInferenceContext(const InferenceContextPtr &inference_context);

Parameters

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>;

Return Value

None

Exception Handling

None

Restrictions

None