import_host_pipeline_ops
功能描述
获得Rec SDK TensorFlow中自定义的TensorFlow算子。
函数原型
1 | def import_host_pipeline_ops(so_pkg_name: str = LIBASC_OPS_SO) -> ModuleType |
参数说明
参数名 |
类型 |
可选/必选 |
说明 |
---|---|---|---|
so_pkg_name |
string |
可选 |
Rec SDK TensorFlow so包名。长度范围:[1, 100] |
返回值说明
- 成功:返回一个包含so中Rec SDK TensorFlow定义的TensorFlow算子的Python封装模块。
- 失败:返回“RuntimeError: when unable to load the library or get the python wrappers.”
使用示例
1 2 | from mx_rec.util.ops import import_host_pipeline_ops host_pipeline_ops = import_host_pipeline_ops() |
父主题: 其他接口