Searched defs:TopKAttr (Results 1 – 1 of 1) sorted by relevance
24582 type TopKAttr func(optionalAttr) type24620 func TopK(scope *Scope, input tf.Output, k int64, optional ...TopKAttr) (values tf.Output, indices …