Searched refs:resultTensor (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/operations/ |
| H A D | Unique.mm | 59 MPSGraphTensor* resultTensor = nil;
79 resultTensor = inputTensor;
87 return {resultTensor, inverseIndicesTensor, countTensor, lengthTensor};
136 resultTensor = [graph scatterWithUpdatesTensor:sortedInput
144 …resultTensor = [graph castTensor:resultTensor toType:[uniqueGraph->inputTensor_ dataType] name:@"c…
178 return {resultTensor, inverseIndicesTensor, countTensor, lengthTensor};
|
| H A D | Distributions.mm | 37 MPSGraphTensor* resultTensor = nil;
102 …newCachedGraph->resultTensor = randomBlock ? randomBlock(newCachedGraph, resultTensors[0]) : resul…
105 …newCachedGraph->resultTensor = castMPSTensor(mpsGraph, newCachedGraph->resultTensor, self.scalar_t…
135 Placeholder outputPlaceholder = Placeholder(cachedGraph->resultTensor, self);
161 MPSGraphTensor* resultTensor = randomTensor;
165 resultTensor = [mpsGraph multiplicationWithPrimaryTensor:randomTensor
171 …return [mpsGraph additionWithPrimaryTensor:resultTensor secondaryTensor:cachedGraph->meanTensor na…
173 return resultTensor;
538 newCachedGraph->resultTensor = [mpsGraph castTensor:reshapeTensor
540 name:@"resultTensor"];
[all …]
|
| H A D | LinearAlgebra.mm | 896 MPSGraphTensor* resultTensor = productTimesAlphaTensor;
898 resultTensor = [mpsGraph additionWithPrimaryTensor:productTimesAlphaTensor
906 newCachedGraph->resultTensor_ = resultTensor;
|
| H A D | Activation.mm | 297 MPSGraphTensor* resultTensor = [mpsGraph subtractionWithPrimaryTensor:gradOutputTensor
303 newCachedGraph->gradInputTensor_ = resultTensor;
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/ |
| H A D | hlo_legalize_to_arithmetic.cc | 99 Value resultTensor = rewriter.create<tensor::FromElementsOp>( in matchAndRewrite() local
103 resultTensor); in matchAndRewrite()
|
| /aosp_15_r20/external/executorch/backends/apple/mps/runtime/operations/ |
| H A D | NormalizationOps.mm | 117 name:@"LayerNorm/resultTensor"];
|
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/ |
| H A D | reducer.hpp | 478 at::Tensor resultTensor; member
|