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