获取当前算子传递infershape推导所需要的关联信息,比如前面算子的shape和DataType信息。
InferenceContextPtr GetInferenceContext() const
无。
参数名 |
类型 |
描述 |
---|---|---|
- |
InferenceContextPtr |
返回当前operator的推理上下文。 InferenceContextPtr是指向InferenceContext类的指针的别名: using InferenceContextPtr = std::shared_ptr<InferenceContext>; |
无。
无。