Searched defs:MPSNodeUnion (Results 1 – 2 of 2) sorted by relevance
330 self, node: torch.fx.Node, mps_graph: MPSGraph, tertiary_op: MPSNodeUnion346 self, node: torch.fx.Node, mps_graph: MPSGraph, binary_op: MPSNodeUnion373 self, node: torch.fx.Node, mps_graph: MPSGraph, unary_op: MPSNodeUnion
673 MPSNodeUnion = Union[ variable