Searched refs:TopKUnique (Results 1 – 7 of 7) sorted by relevance
2 name: "TopKUnique"
5 This is a combination of MakeUnique and TopKUnique. The returned top-K will
2 graph_op_name: "TopKUnique"
204 class TopKUnique : public XlaOpKernel { class206 explicit TopKUnique(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in TopKUnique() function in tensorflow::__anon1472a8820111::TopKUnique227 REGISTER_XLA_OP(Name("TopKUnique"), TopKUnique);
16853 def TF_TopKUniqueOp : TF_Op<"TopKUnique", [NoSideEffect]> {16923 This is a combination of MakeUnique and TopKUnique. The returned top-K will
34990 func TopKUnique(scope *Scope, input tf.Output, k int64) (topk tf.Output, topk_indices tf.Output) { func
56659 name: "TopKUnique"