let summary = "Draws samples from a categorical distribution.";

  let description = [{
The generated values will have a categorical distribution based on the `logits`
or unnormalized log-probabilities provided for all classes.
  }];

  let arguments = (ins
    TFL_FpTensor:$logits,
    TFL_I32Tensor:$num_samples,
    DefaultValuedOptionalAttr<I64Attr, "0">:$seed,

  let summary = "Draws samples from a multinomial distribution.";

  let arguments = (ins
    Arg<TF_IntOrFpTensor, [{2-D Tensor with shape `[batch_size, num_classes]`.  Each slice `[i, :]`
represents the unnormalized log probabilities for all classes.}]>:$logits,
    Arg<TF_Int32Tensor, [{0-D.  Number of independent samples to draw for each row slice.}]>:$num_samples,

    DefaultValuedOptionalAttr<I64Attr, "0">:$seed,