Searched defs:ApproxTopK (Results 1 – 2 of 2) sorted by relevance
157 XlaOp ApproxTopK(XlaBuilder* builder, absl::Span<const XlaOp> operands, in ApproxTopK() function
1151 func ApproxTopK(scope *Scope, input tf.Output, k int64, optional ...ApproxTopKAttr) (values tf.Outp… func