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