Home
last modified time | relevance | path

Searched refs:create_unary_node (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/executorch/backends/apple/mps/operators/
H A Dshape_ops.py46 mps_node = self.create_unary_node(node, mps_graph, MPSPermute)
72 mps_node = self.create_unary_node(node, mps_graph, node_type)
131 mps_node = self.create_unary_node(node, mps_graph, node_type)
159 mps_node = self.create_unary_node(node, mps_graph, MPSSelect)
177 mps_node = self.create_unary_node(node, mps_graph, MPSPixelShuffle)
194 mps_node = self.create_unary_node(node, mps_graph, MPSSlice)
H A Dactivation_ops.py38 mps_node = self.create_unary_node(node, mps_graph, MPSHardTanh)
63 mps_node = self.create_unary_node(node, mps_graph, node_type)
94 mps_node = self.create_unary_node(node, mps_graph, node_type)
H A Dindexing_ops.py38 mps_node = self.create_unary_node(node, mps_graph, MPSIndexSelect)
59 mps_node = self.create_unary_node(node, mps_graph, MPSIndexTensor)
96 mps_node = self.create_unary_node(node, mps_graph, MPSIndexPut)
165 mps_node = self.create_unary_node(node, mps_graph, MPSScatter)
H A Dpad_ops.py32 mps_node = self.create_unary_node(node, mps_graph, MPSConstantPadND)
H A Dreduce_ops.py31 mps_node = self.create_unary_node(node, mps_graph, MPSMean)
H A Dclamp_ops.py33 mps_node = self.create_unary_node(node, mps_graph, MPSClamp)
H A Dconstant_ops.py98 mps_node = self.create_unary_node(node, mps_graph, MPSFullLike)
H A Dunary_ops.py129 self.create_unary_node(node, mps_graph, self.unary_op[node.target])
H A Dnode_visitor.py372 def create_unary_node( member in NodeVisitor