Home
last modified time | relevance | path

Searched refs:create_binary_node (Results 1 – 4 of 4) sorted by relevance

/external/executorch/backends/apple/mps/operators/
Dbinary_ops.py94 mps_node = self.create_binary_node(
155 self.create_binary_node(node, mps_graph, self.comparison_ops[node.target])
Dlinear_algebra_ops.py30 mps_graph.mps_nodes.append(self.create_binary_node(node, mps_graph, MPSMatMul))
Dindexing_ops.py211 mps_node = self.create_binary_node(
Dnode_visitor.py345 def create_binary_node( member in NodeVisitor