向当前算子传递infershape推导所需要的关联信息,比如前面算子的shape和DataType信息。
void SetInferenceContext(const InferenceContextPtr &inference_context)
参数名
输入/输出
描述
inference_context
输入
当前operator的推理上下文。
InferenceContextPtr是指向InferenceContext类的指针的别名:
1
using InferenceContextPtr = std::shared_ptr<InferenceContext>;
无。