(inputs, k)
return top_values, indices
Finds the values and indices of the k largests entries.
Finds the values and indices of the k largests entries.
Finds the values and indices of the k largests entries .
def top_k_with_unique(inputs, k):
  """Finds the values and indices of the k largests entries."""
  unique_inputs = create_make_unique(tf.cast(inputs, tf.float32))
  top_values, indices = create_topk_unique(unique_inputs, k)
  top_values = tf.cast(top_values, inputs.dtype)
  return top_values, indices
def top_k_with_unique ( inputs , k ) : unique_inputs = create_make_unique ( tf . cast ( inputs , tf . float32 ) ) top_values , indices = create_topk_unique ( unique_inputs , k ) top_values = tf . cast ( top_values , inputs . dtype ) return top_values , indices