Home
last modified time | relevance | path

Searched refs:_idToMPSGraphTensor (Results 1 – 18 of 18) sorted by relevance

/external/executorch/backends/apple/mps/runtime/operations/
DShapeOps.mm32 _idToMPSGraphTensor[graphNode->output_id()] = outputTensor;
45 _idToMPSGraphTensor[graphNode->output_id()] =
74 _idToMPSGraphTensor[graphNode->output_id()] =
96 _idToMPSGraphTensor[graphNode->output_id()] =
112 _idToMPSGraphTensor[graphNode->output_id()] =
128 _idToMPSGraphTensor[graphNode->output_id()] =
149 _idToMPSGraphTensor[graphNode->output_id()] =
190 _idToMPSGraphTensor[graphNode->output_id()] = outputTensor;
220 _idToMPSGraphTensor[graphNode->output_id()] =
247 _idToMPSGraphTensor[outId] = mpsGraphResults[crtIdx++];
[all …]
DActivationOps.mm43_idToMPSGraphTensor[graphNode->output_id()] = [_mpsGraph selectWithPredicateTensor:greaterThanMaxP…
60 _idToMPSGraphTensor[graphNode->output_id()] =
88 _idToMPSGraphTensor[graphNode->output_id()] =
89 [_mpsGraph multiplicationWithPrimaryTensor:_idToMPSGraphTensor[graphNode->output_id()]
105 _idToMPSGraphTensor[graphNode->output_id()] =
124 _idToMPSGraphTensor[graphNode->output_id()] =
145 _idToMPSGraphTensor[graphNode->output_id()] =
DConstantOps.mm16 _idToMPSGraphTensor[id] = [_mpsGraph constantWithData:getConstantData(id)
32 _idToMPSGraphTensor[graphNode->output_id()] = nil;
34 _idToMPSGraphTensor[graphNode->output_id()] =
51 _idToMPSGraphTensor[graphNode->output_id()] =
DNormalizationOps.mm59 _idToMPSGraphTensor[graphNode->output1_id()] = [_mpsGraph normalizationWithTensor:inputTensor
70 _idToMPSGraphTensor[graphNode->output2_id()] = [_mpsGraph identityWithTensor:varTensor name:nil];
72_idToMPSGraphTensor[graphNode->output2_id()] = [_mpsGraph identityWithTensor:meanTensor name:nil];
119 _idToMPSGraphTensor[graphNode->output1_id()] = normalizedTensor;
120 _idToMPSGraphTensor[graphNode->output2_id()] = meanTensor;
121 _idToMPSGraphTensor[graphNode->output3_id()] = varianceTensor;
DClampOps.mm38 _idToMPSGraphTensor[graphNode->output_id()] = [_mpsGraph clampWithTensor:inputTensor
47 _idToMPSGraphTensor[graphNode->output_id()] = [_mpsGraph maximumWithPrimaryTensor:inputTensor
55 _idToMPSGraphTensor[graphNode->output_id()] = [_mpsGraph minimumWithPrimaryTensor:inputTensor
83 _idToMPSGraphTensor[graphNode->output_id()] = [_mpsGraph selectWithPredicateTensor:condition
DIndexingOps.mm54 _idToMPSGraphTensor[graphNode->output_id()] =
94 _idToMPSGraphTensor[graphNode->output_id()] =
100 _idToMPSGraphTensor[graphNode->output_id()] = indexSelect(
143 _idToMPSGraphTensor[graphNode->output_id()] =
192 _idToMPSGraphTensor[graphNode->output_id()] =
220 _idToMPSGraphTensor[graphNode->output_id()] =
DUnaryOps.mm33 _idToMPSGraphTensor[graphNode->output_id()] = [_mpsGraph notWithTensor:inputTensor name:nil];
38_idToMPSGraphTensor[graphNode->output_id()] = [_mpsGraph bitwiseNOTWithTensor:inputTensor name:nil…
54 _idToMPSGraphTensor[graphNode->output_id()] = unaryOpTensor( \
127 _idToMPSGraphTensor[graphNode->output_id()] =
DPoolingOps.mm52 _idToMPSGraphTensor[graphNode->output1_id()] = outputs[0];
53 _idToMPSGraphTensor[graphNode->output2_id()] = outputs[1];
100 _idToMPSGraphTensor[graphNode->output1_id()] = avgPoolTensor;
DQuantDequant.mm41 _idToMPSGraphTensor[graphNode->output_id()] = wDqTensor;
43 _idToMPSGraphTensor[graphNode->output_id()] = nil;
DLinearAlgebra.mm26 _idToMPSGraphTensor[graphNode->output_id()] =
76_idToMPSGraphTensor[graphNode->output_id()] = [_mpsGraph additionWithPrimaryTensor:alphaTimesMulti…
DBinaryOps.mm59 _idToMPSGraphTensor[graphNode->output_id()] = binaryOpTensor( \
97 _idToMPSGraphTensor[graphNode->output_id()] = binaryOpTensor( \
127 _idToMPSGraphTensor[graphNode->output_id()] = binaryOpTensor( \
277 _idToMPSGraphTensor[graphNode->output_id()] = divModeTemplate( \
DConvolutionOps.mm68 _idToMPSGraphTensor[graphNode->output_id()] = depthwiseConvTensor;
135 _idToMPSGraphTensor[graphNode->output_id()] = conv2DTensor;
DRangeOps.mm45_idToMPSGraphTensor[graphNode->output_id()] = [_mpsGraph additionWithPrimaryTensor:scaledCoords se…
DReduceOps.mm51 _idToMPSGraphTensor[graphNode->output_id()] = meanTensor;
DPadOps.mm137 _idToMPSGraphTensor[graphNode->output_id()] =
DOperationUtils.mm250 return _idToMPSGraphTensor[id];
/external/executorch/backends/apple/mps/runtime/
DMPSGraphBuilder.mm92 _idToMPSGraphTensor.resize(_flatBufferGraph->mps_values()->size(), nullptr);
120 _idToMPSGraphTensor[out_id] != nil,
124 [_targetTensors addObject: _idToMPSGraphTensor[out_id]];
162 _idToMPSGraphTensor[id] = placeholder;
DMPSGraphBuilder.h180 std::vector<MPSGraphTensor *> _idToMPSGraphTensor; variable