Home
last modified time | relevance | path

Searched refs:getMPSGraphTensor (Results 1 – 25 of 37) sorted by relevance

12

/aosp_15_r20/external/executorch/backends/apple/mps/runtime/operations/
H A DIndexingOps.mm45 MPSGraphTensor* inputTensor = getMPSGraphTensor(graphNode->input1_id());
46 MPSGraphTensor* indexTensor = getMPSGraphTensor(graphNode->index_id());
75 MPSGraphTensor* weightTensor = getMPSGraphTensor(graphNode->input1_id());
76 MPSGraphTensor* indicesTensor = getMPSGraphTensor(graphNode->input2_id());
101 getMPSGraphTensor(graphNode->input1_id()),
103 getMPSGraphTensor(graphNode->input2_id()),
141 MPSGraphTensor* updatesTensor = getMPSGraphTensor(graphNode->input1_id());
142 MPSGraphTensor* indexTensor = getMPSGraphTensor(indexId);
183 MPSGraphTensor* dataTensor = getMPSGraphTensor(graphNode->input1_id());
184 MPSGraphTensor* updatesTensor = getMPSGraphTensor(graphNode->values_id());
[all …]
H A DShapeOps.mm30 _mpsGraph, getMPSGraphTensor(graphNode->input1_id()), permutation
46 [_mpsGraph reshapeTensor:getMPSGraphTensor(graphNode->input1_id())
62 MPSGraphTensor* inputTensor = getMPSGraphTensor(graphNode->input1_id());
92 MPSGraphTensor* catTensor = getMPSGraphTensor(id);
113 [_mpsGraph squeezeTensor:getMPSGraphTensor(graphNode->input1_id())
129 [_mpsGraph expandDimsOfTensor:getMPSGraphTensor(graphNode->input1_id())
144 MPSGraphTensor* slicedTensor = [_mpsGraph sliceTensor:getMPSGraphTensor(graphNode->input1_id())
165 MPSGraphTensor* inputTensor = getMPSGraphTensor(graphNode->input1_id());
202 MPSGraphTensor* inputTensor = getMPSGraphTensor(graphNode->input1_id());
240 mpsGraphResults = [_mpsGraph splitTensor:getMPSGraphTensor(graphNode->input1_id())
[all …]
H A DNormalizationOps.mm30 MPSGraphTensor* inputTensor = getMPSGraphTensor(graphNode->input_id());
31 MPSGraphTensor* meanTensor = getMPSGraphTensor(graphNode->mean_id());
32 MPSGraphTensor* varTensor = getMPSGraphTensor(graphNode->var_id());
33 MPSGraphTensor* weightTensor = getMPSGraphTensor(graphNode->weight_id());
34 MPSGraphTensor* biasTensor = getMPSGraphTensor(graphNode->bias_id());
91 MPSGraphTensor* inputTensor = getMPSGraphTensor(graphNode->input1_id());
92 MPSGraphTensor* weightTensor = getMPSGraphTensor(graphNode->weight_id());
93 MPSGraphTensor* biasTensor = getMPSGraphTensor(graphNode->bias_id());
H A DConvolutionOps.mm48 …MPSGraphTensor* weightTransposeTensor = [_mpsGraph transposeTensor:getMPSGraphTensor(graphNode->in…
52 …hwiseConvTensor = [_mpsGraph depthwiseConvolution3DWithSourceTensor:getMPSGraphTensor(graphNode->i…
59 MPSGraphTensor* biasTensor = getMPSGraphTensor(graphNode->input3_id());
84 MPSGraphTensor* inputTensor = getMPSGraphTensor(graphNode->input1_id());
85 MPSGraphTensor* weightTensor = getMPSGraphTensor(graphNode->input2_id());
120 MPSGraphTensor* biasTensor = getMPSGraphTensor(graphNode->input3_id());
H A DActivationOps.mm25 MPSGraphTensor* inputTensor = getMPSGraphTensor(graphNode->input1_id());
61 [_mpsGraph reLUWithTensor:getMPSGraphTensor(graphNode->input1_id())
90 secondaryTensor:getMPSGraphTensor(graphNode->input1_id())
106 [_mpsGraph leakyReLUWithTensor:getMPSGraphTensor(graphNode->input1_id())
125 [_mpsGraph softMaxWithTensor:getMPSGraphTensor(graphNode->input1_id())
142 …MPSGraphTensor* softmaxTensor = [_mpsGraph softMaxWithTensor:getMPSGraphTensor(graphNode->input1_i…
H A DLinearAlgebra.mm27 [_mpsGraph matrixMultiplicationWithPrimaryTensor:getMPSGraphTensor(graphNode->input1_id())
28 secondaryTensor:getMPSGraphTensor(graphNode->input2_id())
46 MPSGraphTensor* biasTensor = getMPSGraphTensor(graphNode->input1_id());
47 MPSGraphTensor* inputTensor = getMPSGraphTensor(graphNode->input2_id());
48 MPSGraphTensor* weightTensor = getMPSGraphTensor(graphNode->input3_id());
H A DBinaryOps.mm60 getMPSGraphTensor(graphNode->input1_id()), \
61 getMPSGraphTensor(graphNode->input2_id()), \
98 getMPSGraphTensor(graphNode->input1_id()), \
99 getMPSGraphTensor(graphNode->input2_id()), \
128 getMPSGraphTensor(graphNode->input1_id()), \
129 getMPSGraphTensor(graphNode->input2_id()), \
278 getMPSGraphTensor(graphNode->input1_id()), \
279 getMPSGraphTensor(graphNode->input2_id()), \
H A DClampOps.mm25 MPSGraphTensor* inputTensor = getMPSGraphTensor(graphNode->input1_id());
74 MPSGraphTensor* condition = getMPSGraphTensor(graphNode->input1_id());
75 MPSGraphTensor* input = getMPSGraphTensor(graphNode->input2_id());
76 MPSGraphTensor* other = getMPSGraphTensor(graphNode->input3_id());
H A DUnaryOps.mm30 MPSGraphTensor* inputTensor = getMPSGraphTensor(graphNode->input1_id());
55 getMPSGraphTensor(graphNode->input1_id()), \
107 MPSGraphTensor* inputTensor = getMPSGraphTensor(graphNode->input1_id());
H A DQuantDequant.mm31 MPSGraphTensor* inputTensor = getMPSGraphTensor(graphNode->input1_id());
32 MPSGraphTensor* scalesTensor = getMPSGraphTensor(graphNode->scales_id());
H A DPoolingOps.mm47 [_mpsGraph maxPooling2DReturnIndicesWithSourceTensor:getMPSGraphTensor(graphNode->input1_id())
85 …MPSGraphTensor* avgPoolTensor = [_mpsGraph avgPooling2DWithSourceTensor:getMPSGraphTensor(graphNod…
H A DConstantOps.mm53 shape:getMPSGraphTensor(graphNode->input1_id()).shape
H A DReduceOps.mm25 MPSGraphTensor* inputTensor = getMPSGraphTensor(graphNode->input1_id());
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/operations/
H A DRnnOps.mm293 … [feeds setObject:kernelWeight.getMPSGraphTensorData() forKey:kernelWeight.getMPSGraphTensor()];
294 …ct:recurrentKernelWeight.getMPSGraphTensorData() forKey:recurrentKernelWeight.getMPSGraphTensor()];
298 [feeds setObject:bias.getMPSGraphTensorData() forKey:bias.getMPSGraphTensor()];
299 … [feeds setObject:recurrentBias.getMPSGraphTensorData() forKey:recurrentBias.getMPSGraphTensor()];
305 …eeds setObject:selfPlaceholder.getMPSGraphTensorData() forKey:selfPlaceholder.getMPSGraphTensor()];
306 [feeds setObject:selfState.getMPSGraphTensorData() forKey:selfState.getMPSGraphTensor()];
307 … [feeds setObject:selfCellState.getMPSGraphTensorData() forKey:selfCellState.getMPSGraphTensor()];
326 outputPlaceholder0.getMPSGraphTensor() : outputPlaceholder0.getMPSGraphTensorData(),
327 outputPlaceholder1.getMPSGraphTensor() : outputPlaceholder1.getMPSGraphTensorData(),
328 outputPlaceholder2.getMPSGraphTensor() : outputPlaceholder2.getMPSGraphTensorData(),
[all …]
H A DNormalization.mm327 feeds[inputPlaceholder.getMPSGraphTensor()] = inputPlaceholder.getMPSGraphTensorData();
329 feeds[weightPlaceholder.getMPSGraphTensor()] = weightPlaceholder.getMPSGraphTensorData();
331 feeds[biasPlaceholder.getMPSGraphTensor()] = biasPlaceholder.getMPSGraphTensorData();
333 …feeds[runningMeanPlaceholder.getMPSGraphTensor()] = runningMeanPlaceholder.getMPSGraphTensorData();
334 … feeds[runningVarPlaceholder.getMPSGraphTensor()] = runningVarPlaceholder.getMPSGraphTensorData();
338 results[outputPlaceholder.getMPSGraphTensor()] = outputPlaceholder.getMPSGraphTensorData();
339 results[saveMeanPlaceholder.getMPSGraphTensor()] = saveMeanPlaceholder.getMPSGraphTensorData();
340 results[saveVarPlaceholder.getMPSGraphTensor()] = saveVarPlaceholder.getMPSGraphTensorData();
344 results[runningMeanInplaceUpdatePlaceholder.getMPSGraphTensor()] =
346 results[runningVarInplaceUpdatePlaceholder.getMPSGraphTensor()] =
[all …]
H A DPointwiseOps.mm89 selfPlaceholder.getMPSGraphTensor() : selfPlaceholder.getMPSGraphTensorData(),
90 tensor1Placeholder.getMPSGraphTensor() : tensor1Placeholder.getMPSGraphTensorData(),
91 tensor2Placeholder.getMPSGraphTensor() : tensor2Placeholder.getMPSGraphTensorData(),
H A DSummaryOps.mm62 feeds[inputPlaceholder.getMPSGraphTensor()] = inputPlaceholder.getMPSGraphTensorData();
63 feeds[scatterPlaceholder.getMPSGraphTensor()] = scatterPlaceholder.getMPSGraphTensorData();
66 feeds[weightsPlaceholder.getMPSGraphTensor()] = weightsPlaceholder.getMPSGraphTensorData();
H A DLossOps.mm260 feeds[inputPlaceholder.getMPSGraphTensor()] = inputPlaceholder.getMPSGraphTensorData();
261 feeds[targetPlaceholder.getMPSGraphTensor()] = targetPlaceholder.getMPSGraphTensorData();
264 feeds[weightPlaceholder.getMPSGraphTensor()] = weightPlaceholder.getMPSGraphTensorData();
268 … feeds[gradOutputPlaceholder.getMPSGraphTensor()] = gradOutputPlaceholder.getMPSGraphTensorData();
398 feeds[inputPlaceholder.getMPSGraphTensor()] = inputPlaceholder.getMPSGraphTensorData();
399 feeds[targetPlaceholder.getMPSGraphTensor()] = targetPlaceholder.getMPSGraphTensorData();
400 …feeds[totalWeightPlaceholder.getMPSGraphTensor()] = totalWeightPlaceholder.getMPSGraphTensorData();
401 … feeds[gradOutputPlaceholder.getMPSGraphTensor()] = gradOutputPlaceholder.getMPSGraphTensorData();
404 feeds[weightPlaceholder.getMPSGraphTensor()] = weightPlaceholder.getMPSGraphTensorData();
570 feeds[selfPlaceholder.getMPSGraphTensor()] = selfPlaceholder.getMPSGraphTensorData();
[all …]
H A DUnique.mm214 … setObject:outputPlaceholder.getMPSGraphTensorData() forKey:outputPlaceholder.getMPSGraphTensor()];
215 … setObject:lengthPlaceholder.getMPSGraphTensorData() forKey:lengthPlaceholder.getMPSGraphTensor()];
219 forKey:inverseIndicesPlaceholder.getMPSGraphTensor()];
223 … setObject:countsPlaceholder.getMPSGraphTensorData() forKey:countsPlaceholder.getMPSGraphTensor()];
H A DPooling.mm215 feeds[inputPlaceholder.getMPSGraphTensor()] = inputPlaceholder.getMPSGraphTensorData();
216 results[outputPlaceholder.getMPSGraphTensor()] = outputPlaceholder.getMPSGraphTensorData();
219 … feeds[gradOutputPlaceholder.getMPSGraphTensor()] = gradOutputPlaceholder.getMPSGraphTensorData();
223 feeds[indicesPlaceholder.getMPSGraphTensor()] = indicesPlaceholder.getMPSGraphTensorData();
225 … results[indicesPlaceholder.getMPSGraphTensor()] = indicesPlaceholder.getMPSGraphTensorData();
H A DLinear.mm110 feeds[inputPlaceholder.getMPSGraphTensor()] = inputPlaceholder.getMPSGraphTensorData();
111 feeds[weightPlaceholder.getMPSGraphTensor()] = weightPlaceholder.getMPSGraphTensorData();
114 feeds[biasPlaceholder.getMPSGraphTensor()] = biasPlaceholder.getMPSGraphTensorData();
H A DTensorCompare.mm198 feeds[inputPlaceholder.getMPSGraphTensor()] = inputPlaceholder.getMPSGraphTensorData();
204 feeds[minPlaceholder.getMPSGraphTensor()] = minPlaceholder.getMPSGraphTensorData();
211 feeds[maxPlaceholder.getMPSGraphTensor()] = maxPlaceholder.getMPSGraphTensorData();
534 selfPlaceholder.getMPSGraphTensor() : selfPlaceholder.getMPSGraphTensorData(),
H A DBlas.mm178 feeds[matMulVecPlaceholder.getMPSGraphTensor()] = matMulVecPlaceholder.getMPSGraphTensorData();
181 feeds[selfPlaceholder.getMPSGraphTensor()] = selfPlaceholder.getMPSGraphTensorData();
H A DIndexing.mm503 selfPlaceholder.getMPSGraphTensor() : selfPlaceholder.getMPSGraphTensorData(),
504 indexPlaceholder.getMPSGraphTensor() : indexPlaceholder.getMPSGraphTensorData(),
505 sourcePlaceholder.getMPSGraphTensor() : sourcePlaceholder.getMPSGraphTensorData(),
724 selfPlaceholder.getMPSGraphTensor() : selfPlaceholder.getMPSGraphTensorData(),
725 maskPlaceholder.getMPSGraphTensor() : maskPlaceholder.getMPSGraphTensorData(),
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/
H A DOperationUtils.h107 MPSGraphTensor* getMPSGraphTensor() { in getMPSGraphTensor() function
403 return @{ p1.getMPSGraphTensor(): p1.getMPSGraphTensorData() }; in dictionaryFromPlaceholders()
408 p1.getMPSGraphTensor(): p1.getMPSGraphTensorData(), in dictionaryFromPlaceholders()
409 p2.getMPSGraphTensor(): p2.getMPSGraphTensorData(), in dictionaryFromPlaceholders()
415 p1.getMPSGraphTensor(): p1.getMPSGraphTensorData(), in dictionaryFromPlaceholders()
416 p2.getMPSGraphTensor(): p2.getMPSGraphTensorData(), in dictionaryFromPlaceholders()
417 p3.getMPSGraphTensor(): p3.getMPSGraphTensorData(), in dictionaryFromPlaceholders()
423 p1.getMPSGraphTensor(): p1.getMPSGraphTensorData(), in dictionaryFromPlaceholders()
424 p2.getMPSGraphTensor(): p2.getMPSGraphTensorData(), in dictionaryFromPlaceholders()
425 p3.getMPSGraphTensor(): p3.getMPSGraphTensorData(), in dictionaryFromPlaceholders()
[all …]

12