更新时间：2021-03-18 GMT+08:00
GetInferShapeFunc函数
函数功能
获取推导维度回调函数。
函数原型
domi::InferShapeFunc GetInferShapeFunc(const std::string& op_type);
参数说明
参数 |
输入/输出 |
说明 |
|---|---|---|
op_type |
输入 |
算子类型 |
父主题： OpRegistry类
