InferShape
产品支持情况
全量芯片支持。
头文件/库文件
- 头文件：#include <graph/custom_op.h>
- 库文件：libgraph.so
功能说明
形状推理函数，通过形状推理上下文获取输入张量形状信息，推导并设置输出张量的形状，用于图编译阶段的形状推导。
函数原型
1 | virtual graphStatus InferShape(gert::InferShapeContext *ctx) = 0 |
参数说明
参数名 |
输入/输出 |
说明 |
|---|---|---|
ctx |
输入 |
形状推理上下文，可通过上下文获取输入张量形状，设置输出张量形状等。 |
返回值说明
参数名 |
类型 |
说明 |
|---|---|---|
- |
graphStatus |
GRAPH_SUCCESS(0)：成功。 其他值：失败。 |
约束说明
无
父主题： ShapeInferOp