SetInferenceContext
Applicable Products
All processors
Header File/Library File
- Header file: #include <graph/operator.h>
- Library file: libgraph.so
Function Usage
Sets the inference context required for the current operator's InferShape passing, for example, the shape and data type of the upstream operator.
Prototype
1 | 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>; |
Returns
None
Exception Handling
None
Restrictions
None
Parent topic: Operator