Home
last modified time | relevance | path

Searched defs:MPSNodeUnion (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/executorch/backends/apple/mps/operators/
H A Dnode_visitor.py330 self, node: torch.fx.Node, mps_graph: MPSGraph, tertiary_op: MPSNodeUnion
346 self, node: torch.fx.Node, mps_graph: MPSGraph, binary_op: MPSNodeUnion
373 self, node: torch.fx.Node, mps_graph: MPSGraph, unary_op: MPSNodeUnion
/aosp_15_r20/external/executorch/backends/apple/mps/serialization/
H A Dmps_graph_schema.py673 MPSNodeUnion = Union[ variable