import_host_pipeline_ops

功能描述

获得mxrec中自定义的tensorflow算子。

函数原型

def import_host_pipeline_ops(so_pkg_name: str = LIBASC_OPS_SO) -> ModuleType

返回值说明

成功:返回mxrec中自定义的tensorflow算子。

失败:返回“RuntimeError: when unable to load the library or get the python wrappers.”

使用示例

from mx_rec.util.ops import import_host_pipeline_ops
host_pipeline_ops = import_host_pipeline_ops()