Instance Methods

(nonnull instancetype) - initWithDevice:numberOfTopKValues:
(nonnull instancetype) - initWithDevice:
(void) - encodeToCommandBuffer:inputMatrix:resultIndexMatrix:resultValueMatrix:
(nullable instancetype) - initWithCoder:device:
(nonnull instancetype) - copyWithZone:device:

Properties

NSUInteger sourceRows
NSUInteger sourceColumns
NSUInteger indexOffset
NSUInteger numberOfTopKValues

Additional Inherited Members

- (nonnull instancetype) copyWithZone: (nullable NSZone *) zone(nullable id< MTLDevice >) device

Make a copy of this kernel for a new device -

See also:

Parameters:

Returns:

Reimplemented from MPSKernel.

- (void) encodeToCommandBuffer: (nonnull id< MTLCommandBuffer >) commandBuffer(MPSMatrix *__nonnull) inputMatrix(MPSMatrix *__nonnull) resultIndexMatrix(MPSMatrix *__nonnull) resultValueMatrix

Encode a MPSMatrixFindTopK object to a command buffer.

Parameters:

Certain constraints apply to the sizes of the matrices depending on the sizes requested at initialization time as well as the origins at the time this routine is called:

Both result matrices must be large enough to hold a two dimensional array of 'sourceRows' rows and 'numberOfTopKValues' columns beginning at resultMatrixOrigin.

The source matrix must be large enough to contain at least 'numberOfTopKValues' values starting from sourceMatrixOrigin.y.

Each matrix within the range specified by batchStart and batchSize, which also specifies a valid set of matrices within inputMatrix, resultIndexMatrix and resultValueMatrix, will be processed.

The datatypes of the matrices inputMatrix and resultValueMatrix must match and be either MPSDataTypeFloat32 or MPSDataTypeFloat16.

- (nullable instancetype) initWithCoder: (NSCoder *__nonnull) aDecoder(nonnull id< MTLDevice >) device

NSSecureCoding compatability See MPSKernel::initWithCoder.

Parameters:

Returns:

Reimplemented from MPSKernel.

- (nonnull instancetype) initWithDevice: (nonnull id< MTLDevice >) device

Use the above initialization method instead.

Reimplemented from MPSKernel.

- (nonnull instancetype) initWithDevice: (nonnull id< MTLDevice >) device(NSUInteger) numberOfTopKValues

Initialize an MPSMatrixFindTopK object on a device for a given size.

Parameters:

Returns:

- indexOffset [read], [write], [nonatomic], [assign]

Specifies a number that will be added to all the indices written to resultIndexMatrix in encodeToCommandBuffer. This value can be used to offset later computations for example by adding the value for the source matrix column offset sourceMatrixOrigin.y. Example: Let numberOfTopKValues be 3, let the source be the following:

source = [ 6.0, 3.0, 8.0, 1.0, 9.0, 4.0, 5.0 ]

and let the sourceMatrixOrigin.y = 2.

Then if indexOffset = 2 then the result value and result index matrices will be:

result values  = [ 9.0, 8.0, 5.0 ]
result indices = [  4 ,  2 ,  6  ],

which gives the user indices into the original source matrix.

On the other hand if the indexOffset = 0 then the results are as follows:

result values  = [ 9.0, 8.0, 5.0 ]
result indices = [  2 ,  0 ,  4  ],

which on the other hand gives the user indices into the submatrix starting from sourceMatrixOrigin.y == 2.

This property is modifiable and defaults to 0. If a different behavior is desired then this should be modified prior to encoding the kernel.

- numberOfTopKValues [read], [write], [nonatomic], [assign]

The number of highest values (and their indices) to be found in each row by the kernel. This property is initialized in the kernel initialization call initWithDevice, but can be modified before encoding the kernel. Must be less or equal to 16 and requesting more values results in undefined behavior. It is the user's responsibility to ensure that the resultIndexMatrix and resultValueMatrix parameters in encodeToCommandBuffer are large enough to accommodate the results of this operation, otherwise the results of the encode call are undefined.

- sourceColumns [read], [write], [nonatomic], [assign]

The number of columns to consider from the source in the operation. This property is modifiable and defaults to NSUIntegerMax and the number is adjusted dynamically at kernel encode time (see encodeToCommandBuffer) to fit into the source matrix available starting from sourceMatrixOrigin.y, indicating that by default the whole source matrix is used. If a different size is desired then this should be modified prior to encoding the kernel. It is the user's responsibility to ensure that the resultIndexMatrix and resultValueMatrix parameters in encodeToCommandBuffer are large enough to accommodate the results of this operation, otherwise the results of the encode call are undefined. NOTE: sourceMatrixOrigin and resultMatrixOrigin from MPSMatrixUnaryKernel can be used to control the starting points in the source and destination at kernel encode time (see encodeToCommandBuffer).

- sourceRows [read], [write], [nonatomic], [assign]

The number of rows to consider from the source in the operation. This property is modifiable and defaults to NSUIntegerMax and the number is adjusted dynamically at kernel encode time (see encodeToCommandBuffer) to fit into the source matrix available starting from sourceMatrixOrigin.x, indicating that by default the whole source matrix is used. If a different size is desired then this should be modified prior to encoding the kernel. It is the user's responsibility to ensure that the resultIndexMatrix and resultValueMatrix parameters in encodeToCommandBuffer are large enough to accommodate the results of this operation, otherwise the results of the encode call are undefined. NOTE: sourceMatrixOrigin and resultMatrixOrigin from MPSMatrixUnaryKernel can be used to control the starting points in the source and destination at kernel encode time (see encodeToCommandBuffer).