Home
last modified time | relevance | path

Searched refs:MPSBinaryCachedGraph (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/
H A DOperationUtils.h183 struct MPSBinaryCachedGraph : public MPSCachedGraph struct
185 MPSBinaryCachedGraph(MPSGraph *graph) : MPSCachedGraph(graph) {} in MPSBinaryCachedGraph() argument
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/operations/
H A DEye.mm62 …// consider using `MPSUnaryCachedGraph` or `MPSBinaryCachedGraph` and their corresponding Grad ver…
H A DBlas.mm57 using CachedGraph = MPSBinaryCachedGraph;
H A DReduceOps.mm44 typedef MPSGraphTensor* (^NormOpBlock)(MPSBinaryCachedGraph*, MPSGraphTensor*, MPSGraphTensor*);
46 …MPSGraphTensor*(MPSBinaryCachedGraph * graph, MPSGraphTensor * primary, MPSGraphTensor * secondary)
355 …auto cachedGraph = LookUpOrCreateCachedGraph<MPSBinaryCachedGraph>(key, [&](auto mpsGraph, auto ne…
H A DTensorCompare.mm306 …auto cachedGraph = LookUpOrCreateCachedGraph<MPSBinaryCachedGraph>(key, [&](auto mpsGraph, auto ne…
H A DLinearAlgebra.mm252 using CachedGraph = MPSBinaryCachedGraph;