Home
last modified time | relevance | path

Searched refs:TopKWithUnique (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DTopKWithUnique.pbtxt2 name: "TopKWithUnique"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_TopKWithUnique.pbtxt2 graph_op_name: "TopKWithUnique"
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtopk_ops.cc334 class TopKWithUnique : public XlaOpKernel { class
336 explicit TopKWithUnique(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in TopKWithUnique() function in tensorflow::__anon1472a8820111::TopKWithUnique
358 REGISTER_XLA_OP(Name("TopKWithUnique"), TopKWithUnique);
/external/tensorflow/tensorflow/go/op/
Dwrappers.go20200 func TopKWithUnique(scope *Scope, input tf.Output, k int64) (topk tf.Output, topk_indices tf.Output… func
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_generated_ops.td16919 def TF_TopKWithUniqueOp : TF_Op<"TopKWithUnique", [NoSideEffect]> {
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt56724 name: "TopKWithUnique"