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