Home
last modified time | relevance | path

Searched refs:getMPSGraphTensorData (Results 1 – 22 of 22) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/operations/
H A DRnnOps.mm293 … [feeds setObject:kernelWeight.getMPSGraphTensorData() forKey:kernelWeight.getMPSGraphTensor()];
294 …[feeds setObject:recurrentKernelWeight.getMPSGraphTensorData() forKey:recurrentKernelWeight.getMPS…
298 [feeds setObject:bias.getMPSGraphTensorData() forKey:bias.getMPSGraphTensor()];
299 … [feeds setObject:recurrentBias.getMPSGraphTensorData() forKey:recurrentBias.getMPSGraphTensor()];
305 …[feeds 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();
345 runningMeanInplaceUpdatePlaceholder.getMPSGraphTensorData();
347 runningVarInplaceUpdatePlaceholder.getMPSGraphTensorData();
[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 …[results setObject:outputPlaceholder.getMPSGraphTensorData() forKey:outputPlaceholder.getMPSGraphT…
215 …[results setObject:lengthPlaceholder.getMPSGraphTensorData() forKey:lengthPlaceholder.getMPSGraphT…
218 [results setObject:inverseIndicesPlaceholder.getMPSGraphTensorData()
223 …[results setObject:countsPlaceholder.getMPSGraphTensorData() forKey:countsPlaceholder.getMPSGraphT…
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(),
H A DConvolution.mm345 feeds[inputPlaceholder.getMPSGraphTensor()] = inputPlaceholder.getMPSGraphTensorData();
346 feeds[weightsPlaceholder.getMPSGraphTensor()] = weightsPlaceholder.getMPSGraphTensorData();
348 feeds[biasPlaceholder.getMPSGraphTensor()] = biasPlaceholder.getMPSGraphTensorData();
H A DConstantOps.mm53 @{outputPlaceholder.getMPSGraphTensor() : outputPlaceholder.getMPSGraphTensorData()};
H A DPad.mm309 feeds[inputPlaceholder.getMPSGraphTensor()] = inputPlaceholder.getMPSGraphTensorData();
311 … feeds[gradOutputPlaceholder.getMPSGraphTensor()] = gradOutputPlaceholder.getMPSGraphTensorData();
H A DDistributions.mm127 feeds[stdPlaceholder.getMPSGraphTensor()] = stdPlaceholder.getMPSGraphTensorData();
132 feeds[meanPlaceholder.getMPSGraphTensor()] = meanPlaceholder.getMPSGraphTensorData();
559 probPlaceholder.getMPSGraphTensor() : probPlaceholder.getMPSGraphTensorData()
H A DActivation.mm1185 selfPlaceholder.getMPSGraphTensor() : selfPlaceholder.getMPSGraphTensorData(),
1261 gradOutputPlaceholder.getMPSGraphTensor() : gradOutputPlaceholder.getMPSGraphTensorData(),
1262 selfPlaceholder.getMPSGraphTensor() : selfPlaceholder.getMPSGraphTensorData(),
1442 selfPlaceholder.getMPSGraphTensor() : selfPlaceholder.getMPSGraphTensorData(),
1508 gradOutputPlaceholder.getMPSGraphTensor() : gradOutputPlaceholder.getMPSGraphTensorData(),
1509 selfPlaceholder.getMPSGraphTensor() : selfPlaceholder.getMPSGraphTensorData(),
H A DBinaryOps.mm162 feeds[selfPlaceholder.getMPSGraphTensor()] = selfPlaceholder.getMPSGraphTensorData();
173 feeds[otherPlaceholder.getMPSGraphTensor()] = otherPlaceholder.getMPSGraphTensorData();
H A DUpSample.mm208 inputPlaceholder.getMPSGraphTensor() : inputPlaceholder.getMPSGraphTensorData(),
H A DShape.mm319 feeds[inputPlaceholder.getMPSGraphTensor()] = inputPlaceholder.getMPSGraphTensorData();
H A DReduceOps.mm416 feeds[inputPlaceholder.getMPSGraphTensor()] = inputPlaceholder.getMPSGraphTensorData();
420 feeds[otherPlaceholder.getMPSGraphTensor()] = otherPlaceholder.getMPSGraphTensorData();
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/
H A DOperationUtils.h110 MPSGraphTensorData* getMPSGraphTensorData() { in getMPSGraphTensorData() function
129 MPSGraphTensorData *getMPSGraphTensorData(MPSGraph* mpsGraph, MPSStream* mpsStream, const Tensor& t…
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()
[all …]
H A DOperationUtils.mm585 MPSGraphTensorData* getMPSGraphTensorData(MPSGraph* mpsGraph, MPSStream* mpsStream, const Tensor& t…