GetInputHandleShapesAndTypes
产品支持情况
产品 |
是否支持 |
|---|---|
Atlas 350 加速卡 |
√ |
√ |
|
√ |
|
√ |
|
√ |
|
√ |
头文件/库文件
- 头文件:#include <graph/inference_context.h>
- 库文件:libgraph.so
功能说明
在推理上下文中,获取算子输入句柄的ShapeAndType。
函数原型
1 | const std::vector<std::vector<ShapeAndType>> &GetInputHandleShapesAndTypes() const |
参数说明
无。
返回值说明
算子输入句柄的ShapeAndType。
异常处理
无。
约束说明
无。
父主题: InferenceContext