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