Searched refs:TopKShapeFn (Results 1 – 1 of 1) sorted by relevance
1199 Status TopKShapeFn(InferenceContext* c) { in TopKShapeFn() function1243 .SetShapeFn(TopKShapeFn);1253 .SetShapeFn(TopKShapeFn);