rml_op_top_k_params

Parameters for the RML_OP_TOP_K operation.

API

Graph Manipulation

Items

Name	Type	Description
input	rml_op	Operation that outputs tensor containing input data.
axis	int32_t	Dimension on which to do the sort. Negative value means counting dimensions from the back. Accepted range is [-r, r-1] where r = rank(input).
k	uint32_t	Number of returned elements.