MCPcopy Create free account
hub / github.com/pytorch/executorch / get_node_visitors

Function get_node_visitors

backends/apple/mps/operators/node_visitor.py:397–411  ·  view source on GitHub ↗
(*args)

Source from the content-addressed store, hash-verified

395
396
397def get_node_visitors(*args) -> Dict[str, NodeVisitor]:
398 node_visitors = {}
399 """
400 Create a new class instance at runtime, and put them in a dict
401 """
402 for target, visitor in _node_visitor_dict.items():
403 assert callable(
404 visitor
405 ), f"Expecting a callable class, but got {visitor} of type {type(visitor)}"
406 node_visitors[target] = visitor(*args)
407
408 placeholder_output_visitor = NodeVisitor(*args)
409 node_visitors["placeholder"] = placeholder_output_visitor
410 node_visitors["output"] = placeholder_output_visitor
411 return node_visitors
412
413
414def process_placeholder_nodes(

Callers 2

preprocessMethod · 0.90
__init__Method · 0.90

Calls 2

itemsMethod · 0.80
NodeVisitorClass · 0.70

Tested by

no test coverage detected