Searched defs:TopKWithUnique (Results 1 – 2 of 2) sorted by relevance
334 class TopKWithUnique : public XlaOpKernel { class336 explicit TopKWithUnique(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in TopKWithUnique() function in tensorflow::__anon5f1cb7a70111::TopKWithUnique
36764 func TopKWithUnique(scope *Scope, input tf.Output, k int64) (topk tf.Output, topk_indices tf.Output… func