ACGPushOpsToDatasetHook

该Hook用于支持经由Constant、String等类型OP的输出Tensor作为稀疏表查询的参数。通过改图的方式将其子图推入到Dataset的Map函数内执行，达到操作流程切换CPU的效果。

>目前仅在使用NPUEstimator训练模式下使用，且必须添加在GraphModifierHook之前。当前仅支持Tensorflow 1.0版本，不支持控制依赖的和共享变量等复杂OP。

参数名	参数类型	参数说明	必选/可选
dump_graph	bool	是否保存Tensorflow当前计算图，默认为“False”。	可选

使用示例

from mx_rec.graph.modifier import GraphModifierHook
from mx_rec.graph.acg_push_ops import ACGPushOpsToDatasetHook
def input_fn():
    pass
hooks_list = [ACGPushOpsToDatasetHook(dump_graph=True), GraphModifierHook(modify_graph=params.modify_graph)]
est = NPUEstimator(...)
est.train(input_fn=lambda: input_fn, hooks=npu_hooks_append(hooks_list))

父主题： 类参考