Home
last modified time | relevance | path

Searched defs:MPSGraph (Results 1 – 23 of 23) sorted by relevance

/aosp_15_r20/external/executorch/backends/apple/mps/operators/
H A Dnode_visitor.py67 mps_graph: MPSGraph,
74 mps_graph: MPSGraph,
111 def define_tensor_list(self, node: torch.fx.Node, mps_graph: MPSGraph) -> List[int]:
153 mps_graph: MPSGraph,
189 mps_graph: MPSGraph,
222 mps_graph: MPSGraph,
254 mps_graph: MPSGraph,
294 self, node: Union[torch.fx.Node, float, int], mps_graph: MPSGraph, hash=None
330 self, node: torch.fx.Node, mps_graph: MPSGraph, tertiary_op: MPSNodeUnion
346 self, node: torch.fx.Node, mps_graph: MPSGraph, binary_op: MPSNodeUnion
[all …]
H A Dshape_ops.py44 mps_graph: MPSGraph,
65 mps_graph: MPSGraph,
91 mps_graph: MPSGraph,
123 mps_graph: MPSGraph,
157 mps_graph: MPSGraph,
175 mps_graph: MPSGraph,
192 mps_graph: MPSGraph,
239 mps_graph: MPSGraph,
H A Dindexing_ops.py36 mps_graph: MPSGraph,
57 mps_graph: MPSGraph,
94 mps_graph: MPSGraph,
163 mps_graph: MPSGraph,
208 mps_graph: MPSGraph,
H A Dop_quant_dequant.py36 mps_graph: MPSGraph,
87 mps_graph: MPSGraph,
112 mps_graph: MPSGraph,
134 mps_graph: MPSGraph,
H A Dactivation_ops.py36 mps_graph: MPSGraph,
60 mps_graph: MPSGraph,
87 mps_graph: MPSGraph,
H A Dpooling_ops.py32 mps_graph: MPSGraph,
91 mps_graph: MPSGraph,
H A Dlinear_algebra_ops.py28 mps_graph: MPSGraph,
43 mps_graph: MPSGraph,
H A Dnormalization_ops.py34 mps_graph: MPSGraph,
74 mps_graph: MPSGraph,
H A Dclamp_ops.py31 mps_graph: MPSGraph,
58 mps_graph: MPSGraph,
H A Dconstant_ops.py43 mps_graph: MPSGraph,
92 mps_graph: MPSGraph,
H A Dbinary_ops.py92 mps_graph: MPSGraph,
151 mps_graph: MPSGraph,
H A Dop_skip_ops.py22 mps_graph: MPSGraph,
H A Dpad_ops.py30 mps_graph: MPSGraph,
H A Dop_getitem.py25 mps_graph: MPSGraph,
H A Drange_ops.py29 mps_graph: MPSGraph,
H A Dconvolution_ops.py32 mps_graph: MPSGraph,
H A Dreduce_ops.py29 mps_graph: MPSGraph,
H A Dop_clone.py26 mps_graph: MPSGraph,
H A Dunary_ops.py126 mps_graph: MPSGraph,
/aosp_15_r20/external/executorch/backends/apple/mps/
H A Dmps_preprocess.py171 mps_graph: MPSGraph,
197 mps_graph: MPSGraph,
207 mps_graph: MPSGraph,
218 mps_graph: MPSGraph,
231 def _extract_constant_segment(mps_graph: MPSGraph) -> Tuple[Cord, MPSGraph]:
263 def pretty_print(mps_graph: MPSGraph):
/aosp_15_r20/external/executorch/backends/apple/mps/serialization/
H A Dmps_graph_serialize.py17 def convert_to_flatbuffer(mps_graph: MPSGraph) -> bytes:
H A Dmps_graph_schema.py809 class MPSGraph: class
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/
H A DOperationUtils.h300 inline T* LookUpOrCreateCachedGraph(const std::string& key, std::function<void(MPSGraph*, T*)> inst… in LookUpOrCreateCachedGraph()