SetInferenceContext
Applicability
Product |
Supported |
|---|---|
Atlas 350 Accelerator Card |
√ |
√ |
|
√ |
|
√ |
|
√ |
|
√ |
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.
|
Returns
None
Exception Handling
None
Restrictions
None
Parent topic: Operator