昇腾社区首页
中文
注册
开发者
下载

简介

GE改图相关的工具类。如shape推导,校验节点是否支持,部分图优化工具方法等。

需要包含的头文件

1
#include <ge/ge_utils.h>

Public成员函数

static Status InferShape(const Graph &graph, const std::vector<Shape> &input_shape)
static Status CheckNodeSupportOnAicore(const GNode &node, bool &is_supported, AscendString &unsupported_reason)