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