该Hook用于支持稀疏表查询时传入向上无法找到Dataset的孤儿Key类型,主要用于拓展自动改图模式下的稀疏表查询功能。如果没有找到目标类型的算子,不会执行切分操作。
该Hook的使用场景是NPUEstimator模式,启用自动改图功能。该Hook的调用时机需要在自动改图的GraphModifierHook之前。
1 2 3 4 5 6 7 8 9 10 11 12 13 | from mx_rec.graph import OrphanLookupKeySlicerHook, GraphModifierHook def input_fn(): """ 用户自定义Estimator输入函数。 """ orphan_slicer_hook = OrphanLookupKeySlicerHook() modifier_hook = GraphModifierHook(modify_graph=params.modify_graph) hooks_list = [orphan_slicer_hook, modifier_hook] est = NPUEstimator(...) est.train(input_fn=lambda: input_fn, hooks=npu_hooks_append(hooks_list)) |