morph-net
morph-net copied to clipboard
Fixing a small issue in OpRegularizerManager.
Fixing a small issue in OpRegularizerManager.
Internally, OpRegularizerManager was passing output_boundary as a list to _dfs_for_source_ops, which was packaging it again as a list. The fix now allows _dfs_for_source_ops to support a single output_boundary node or a list of nodes.