Searched refs:TopKAttr (Results 1 – 1 of 1) sorted by relevance
22060 type TopKAttr func(optionalAttr) type22067 func TopKSorted(value bool) TopKAttr {22096 func TopK(scope *Scope, input tf.Output, k int64, optional ...TopKAttr) (values tf.Output, indices … argument