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