Searched refs:TopKWithUnique (Results 1 – 6 of 6) sorted by relevance
2 name: "TopKWithUnique"
2 graph_op_name: "TopKWithUnique"
334 class TopKWithUnique : public XlaOpKernel { class336 explicit TopKWithUnique(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in TopKWithUnique() function in tensorflow::__anon1472a8820111::TopKWithUnique358 REGISTER_XLA_OP(Name("TopKWithUnique"), TopKWithUnique);
20200 func TopKWithUnique(scope *Scope, input tf.Output, k int64) (topk tf.Output, topk_indices tf.Output… func
16919 def TF_TopKWithUniqueOp : TF_Op<"TopKWithUnique", [NoSideEffect]> {
56724 name: "TopKWithUnique"