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::__anonee83e3f10111::TopKUnique
53020 func TopKUnique(scope *Scope, input tf.Output, k int64) (topk tf.Output, topk_indices tf.Output) { func