-
-
Methods in org.robovm.apple.metalps with parameters of type NSMutableArray
| 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) |
|
MPSImage |
MPSNNGraph.encode(MTLCommandBuffer commandBuffer,
NSArray<MPSImage> sourceImages,
NSArray<MPSState> sourceStates,
NSMutableArray<MPSImage> intermediateImages,
NSMutableArray<MPSState> destinationStates) |
|
NSArray<MPSImage> |
MPSNNGraph.encodeBatch(MTLCommandBuffer commandBuffer,
NSArray<NSArray<MPSImage>> sourceImages,
NSArray<NSArray<? extends MPSState>> sourceStates,
NSMutableArray<NSArray<MPSImage>> intermediateImages,
NSMutableArray<NSArray<? extends MPSState>> destinationStates) |
|
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,
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 |
MPSRNNImageInferenceLayer.encodeSequence(MTLCommandBuffer commandBuffer,
NSArray<MPSImage> sourceImages,
NSArray<MPSImage> destinationImages,
MPSRNNRecurrentImageState recurrentInputState,
NSMutableArray<MPSRNNRecurrentImageState> recurrentOutputStates) |
|
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 |
MPSRNNMatrixTrainingLayer.init(MTLDevice device,
MPSRNNDescriptor rnnDescriptor,
NSMutableArray<MPSMatrix> trainableWeights) |
|
-
-