SetInferenceContext
Description
Sets the inference context for inferShape passing to the current operator, for example, the shape and data type of the upstream operator.
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.
|
Returns
None
Exception Handling
None
Restrictions
None
Parent topic: Operator