-
Methods in org.robovm.apple.metalps with parameters of type MPSMatrix
| Modifier and Type |
Method |
Description |
static MPSMatrixCopyDescriptor |
MPSMatrixCopyDescriptor.create(MPSMatrix sourceMatrix,
MPSMatrix destinationMatrix,
MPSMatrixCopyOffsets offsets) |
|
void |
MPSImageCopyToMatrix.encode(MTLCommandBuffer commandBuffer,
NSArray<MPSImage> sourceImages,
MPSMatrix destinationMatrix) |
|
void |
MPSImageCopyToMatrix.encode(MTLCommandBuffer commandBuffer,
MPSImage sourceImage,
MPSMatrix destinationMatrix) |
|
void |
MPSMatrixBatchNormalization.encode(MTLCommandBuffer commandBuffer,
MPSMatrix inputMatrix,
MPSVector meanVector,
MPSVector varianceVector,
MPSVector gammaVector,
MPSVector betaVector,
MPSMatrix resultMatrix) |
|
void |
MPSMatrixBatchNormalizationGradient.encode(MTLCommandBuffer commandBuffer,
MPSMatrix gradientMatrix,
MPSMatrix inputMatrix,
MPSVector meanVector,
MPSVector varianceVector,
MPSVector gammaVector,
MPSVector betaVector,
MPSMatrix resultGradientForDataMatrix,
MPSVector resultGradientForGammaVector,
MPSVector resultGradientForBetaVector) |
|
void |
MPSMatrixCopyToImage.encode(MTLCommandBuffer commandBuffer,
MPSMatrix sourceMatrix,
MPSImage destinationImage) |
|
void |
MPSMatrixDecompositionCholesky.encode(MTLCommandBuffer commandBuffer,
MPSMatrix sourceMatrix,
MPSMatrix resultMatrix,
MTLBuffer status) |
|
void |
MPSMatrixDecompositionLU.encode(MTLCommandBuffer commandBuffer,
MPSMatrix sourceMatrix,
MPSMatrix resultMatrix,
MPSMatrix pivotIndices,
MTLBuffer status) |
|
void |
MPSMatrixFindTopK.encode(MTLCommandBuffer commandBuffer,
MPSMatrix inputMatrix,
MPSMatrix resultIndexMatrix,
MPSMatrix resultValueMatrix) |
|
void |
MPSMatrixFullyConnected.encode(MTLCommandBuffer commandBuffer,
MPSMatrix inputMatrix,
MPSMatrix weightMatrix,
MPSVector biasVector,
MPSMatrix resultMatrix) |
|
void |
MPSMatrixMultiplication.encode(MTLCommandBuffer commandBuffer,
MPSMatrix leftMatrix,
MPSMatrix rightMatrix,
MPSMatrix resultMatrix) |
|
void |
MPSMatrixNeuron.encode(MTLCommandBuffer commandBuffer,
MPSMatrix inputMatrix,
MPSVector biasVector,
MPSMatrix resultMatrix) |
|
void |
MPSMatrixNeuronGradient.encode(MTLCommandBuffer commandBuffer,
MPSMatrix gradientMatrix,
MPSMatrix inputMatrix,
MPSVector biasVector,
MPSMatrix resultGradientForDataMatrix,
MPSVector resultGradientForBiasVector) |
|
void |
MPSMatrixRandom.encode(MTLCommandBuffer commandBuffer,
MPSMatrix destinationMatrix) |
|
void |
MPSMatrixSoftMax.encode(MTLCommandBuffer commandBuffer,
MPSMatrix inputMatrix,
MPSMatrix resultMatrix) |
|
void |
MPSMatrixSoftMaxGradient.encode(MTLCommandBuffer commandBuffer,
MPSMatrix gradientMatrix,
MPSMatrix forwardOutputMatrix,
MPSMatrix resultMatrix) |
|
void |
MPSMatrixSolveCholesky.encode(MTLCommandBuffer commandBuffer,
MPSMatrix sourceMatrix,
MPSMatrix rightHandSideMatrix,
MPSMatrix solutionMatrix) |
|
void |
MPSMatrixSolveLU.encode(MTLCommandBuffer commandBuffer,
MPSMatrix sourceMatrix,
MPSMatrix rightHandSideMatrix,
MPSMatrix pivotIndices,
MPSMatrix solutionMatrix) |
|
void |
MPSMatrixSolveTriangular.encode(MTLCommandBuffer commandBuffer,
MPSMatrix sourceMatrix,
MPSMatrix rightHandSideMatrix,
MPSMatrix solutionMatrix) |
|
void |
MPSMatrixSum.encode(MTLCommandBuffer buffer,
NSArray<MPSMatrix> sourceMatrices,
MPSMatrix resultMatrix,
MPSVector scaleVector,
MPSVector offsetVector,
MPSVector biasVector,
long startIndex) |
|
void |
MPSMatrixVectorMultiplication.encode(MTLCommandBuffer commandBuffer,
MPSMatrix inputMatrix,
MPSVector inputVector,
MPSVector resultVector) |
|
void |
MPSNNOptimizerAdam.encode(MTLCommandBuffer commandBuffer,
MPSMatrix inputGradientMatrix,
MPSMatrix inputValuesMatrix,
MPSMatrix inputMomentumMatrix,
MPSMatrix inputVelocityMatrix,
MPSMatrix resultValuesMatrix) |
|
void |
MPSNNOptimizerAdam.encode(MTLCommandBuffer commandBuffer,
MPSMatrix inputGradientMatrix,
MPSMatrix inputValuesMatrix,
MPSMatrix inputMomentumMatrix,
MPSMatrix inputVelocityMatrix,
MPSMatrix maximumVelocityMatrix,
MPSMatrix resultValuesMatrix) |
|
void |
MPSNNOptimizerRMSProp.encode(MTLCommandBuffer commandBuffer,
MPSMatrix inputGradientMatrix,
MPSMatrix inputValuesMatrix,
MPSMatrix inputSumOfSquaresMatrix,
MPSMatrix resultValuesMatrix) |
|
void |
MPSNNOptimizerStochasticGradientDescent.encode(MTLCommandBuffer commandBuffer,
MPSMatrix inputGradientMatrix,
MPSMatrix inputValuesMatrix,
MPSMatrix inputMomentumMatrix,
MPSMatrix resultValuesMatrix) |
|
void |
MPSMatrixCopyToImage.encodeBatch(MTLCommandBuffer commandBuffer,
MPSMatrix sourceMatrix,
NSArray<MPSImage> destinationImages) |
|
void |
MPSRNNMatrixTrainingLayer.encodeCopyWeights(MTLCommandBuffer commandBuffer,
NSArray<MPSMatrix> weights,
MPSRNNMatrixId matrixId,
MPSMatrix matrix,
boolean copyFromWeightsToMatrix,
MTLOrigin matrixOffset) |
|
void |
MPSMatrixFullyConnectedGradient.encodeGradient(MTLCommandBuffer commandBuffer,
MPSMatrix gradientMatrix,
MPSMatrix weightMatrix,
MPSMatrix resultGradientForDataMatrix) |
|
void |
MPSMatrixFullyConnectedGradient.encodeGradient(MTLCommandBuffer commandBuffer,
MPSMatrix gradientMatrix,
MPSMatrix inputMatrix,
MPSMatrix resultGradientForWeightMatrix,
MPSVector resultGradientForBiasVector) |
|
void |
MPSMatrixCopyDescriptor.setCopyOperation(long index,
MPSMatrix sourceMatrix,
MPSMatrix destinationMatrix,
MPSMatrixCopyOffsets offsets) |
|
Method parameters in org.robovm.apple.metalps with type arguments of type MPSMatrix
| Modifier and Type |
Method |
Description |
void |
MPSRNNMatrixTrainingLayer.createTemporaryWeightGradientMatrices(NSMutableArray<MPSMatrix> matricesOut,
MPSDataType dataType,
MTLCommandBuffer commandBuffer) |
|
void |
MPSRNNMatrixTrainingLayer.createWeightGradientMatrices(NSMutableArray<MPSMatrix> matricesOut,
MPSDataType dataType) |
|
void |
MPSRNNMatrixTrainingLayer.createWeightMatrices(NSMutableArray<MPSMatrix> matricesOut) |
|
void |
MPSMatrixSum.encode(MTLCommandBuffer buffer,
NSArray<MPSMatrix> sourceMatrices,
MPSMatrix resultMatrix,
MPSVector scaleVector,
MPSVector offsetVector,
MPSVector biasVector,
long startIndex) |
|
void |
MPSRNNMatrixInferenceLayer.encodeBidirectionalSequence(MTLCommandBuffer commandBuffer,
NSArray<MPSMatrix> sourceSequence,
NSArray<MPSMatrix> destinationForwardMatrices,
NSArray<MPSMatrix> destinationBackwardMatrices) |
|
void |
MPSRNNMatrixTrainingLayer.encodeCopyWeights(MTLCommandBuffer commandBuffer,
NSArray<MPSMatrix> weights,
MPSRNNMatrixId matrixId,
MPSMatrix matrix,
boolean copyFromWeightsToMatrix,
MTLOrigin matrixOffset) |
|
void |
MPSRNNMatrixTrainingLayer.encodeForwardSequence(MTLCommandBuffer commandBuffer,
NSArray<MPSMatrix> sourceMatrices,
NSArray<MPSMatrix> destinationMatrices,
NSMutableArray<MPSRNNMatrixTrainingState> trainingStates,
NSArray<MPSMatrix> weights) |
|
void |
MPSRNNMatrixTrainingLayer.encodeForwardSequence(MTLCommandBuffer commandBuffer,
NSArray<MPSMatrix> sourceMatrices,
MachineSizedUIntPtr sourceOffsets,
NSArray<MPSMatrix> destinationMatrices,
MachineSizedUIntPtr destinationOffsets,
NSMutableArray<MPSRNNMatrixTrainingState> trainingStates,
MPSRNNRecurrentMatrixState recurrentInputState,
NSMutableArray<MPSRNNRecurrentMatrixState> recurrentOutputStates,
NSArray<MPSMatrix> weights) |
|
void |
MPSRNNMatrixTrainingLayer.encodeGradientSequence(MTLCommandBuffer commandBuffer,
NSArray<MPSMatrix> forwardSources,
NSArray<MPSMatrix> sourceGradients,
NSArray<MPSMatrix> destinationGradients,
NSArray<MPSMatrix> weightGradients,
NSArray<MPSRNNMatrixTrainingState> trainingStates,
NSArray<MPSMatrix> weights) |
|
void |
MPSRNNMatrixTrainingLayer.encodeGradientSequence(MTLCommandBuffer commandBuffer,
NSArray<MPSMatrix> forwardSources,
MachineSizedUIntPtr forwardSourceOffsets,
NSArray<MPSMatrix> sourceGradients,
MachineSizedUIntPtr sourceGradientOffsets,
NSArray<MPSMatrix> destinationGradients,
MachineSizedUIntPtr destinationOffsets,
NSArray<MPSMatrix> weightGradients,
NSArray<MPSRNNMatrixTrainingState> trainingStates,
MPSRNNRecurrentMatrixState recurrentInputState,
NSMutableArray<MPSRNNRecurrentMatrixState> recurrentOutputStates,
NSArray<MPSMatrix> weights) |
|
void |
MPSRNNMatrixInferenceLayer.encodeSequence(MTLCommandBuffer commandBuffer,
NSArray<MPSMatrix> sourceMatrices,
NSArray<MPSMatrix> destinationMatrices,
MPSRNNRecurrentMatrixState recurrentInputState,
NSMutableArray<MPSRNNRecurrentMatrixState> recurrentOutputStates) |
|
void |
MPSRNNMatrixInferenceLayer.encodeSequence(MTLCommandBuffer commandBuffer,
NSArray<MPSMatrix> sourceMatrices,
MachineSizedUIntPtr sourceOffsets,
NSArray<MPSMatrix> destinationMatrices,
MachineSizedUIntPtr destinationOffsets,
MPSRNNRecurrentMatrixState recurrentInputState,
NSMutableArray<MPSRNNRecurrentMatrixState> recurrentOutputStates) |
|
protected long |
MPSMatrixCopyDescriptor.init(NSArray<MPSMatrix> sourceMatrices,
NSArray<MPSMatrix> destinationMatrices,
MPSVector offsets,
long byteOffset) |
|
protected long |
MPSRNNMatrixTrainingLayer.init(MTLDevice device,
MPSRNNDescriptor rnnDescriptor,
NSMutableArray<MPSMatrix> trainableWeights) |
|
-