跳到主要内容

TryGetInputDesc

函数功能

根据算子Input名称获取算子Input的TensorDesc。

函数原型

数据类型为string的接口后续版本会废弃,建议使用数据类型为非string的接口。

graphStatus TryGetInputDesc(const std::string &name, TensorDesc &tensor_desc) const;
graphStatus TryGetInputDesc(const char_t *name, TensorDesc &tensor_desc) const;

参数说明

参数名输入/输出描述
name输入算子的Input名。
tensor_desc输出返回算子端口的当前设置格式,为TensorDesc对象。

返回值

类型描述
graphStatustrue:有此端口,获取TensorDesc成功。 false:无此端口,出参为空,获取TensorDesc失败。

异常处理

异常场景说明
无对应name输入返回false。

约束说明