昇腾社区首页
中文
注册

简介

CtInferShapeContext继承自InferShapeContext,用于编译期Shape推导,提供编译期使用的接口。

该类主要用于资源类算子,资源类算子分为创建资源的算子和使用资源的算子两类,这两类算子通过传递DT_RESOURCE句柄来关联资源。句柄信息通过CtInferShapeContext提供的接口在编译期获取。

CtInferShapeContext继承关系图如下:

KernelContext类

需要包含的头文件

1
#include <ct_infer_shape_context.h>

Public成员函数

ge::InferenceContext *GetInferenceContext() const