op_AU      uint32 = 0x7E00 // FORMAT_RX1        ADD UNNORMALIZED (short HFP)
	op_AUR     uint32 = 0x3E00 // FORMAT_RR         ADD UNNORMALIZED (short HFP)
	op_AW      uint32 = 0x6E00 // FORMAT_RX1        ADD UNNORMALIZED (long HFP)
	op_AWR     uint32 = 0x2E00 // FORMAT_RR         ADD UNNORMALIZED (long HFP)
	op_AXBR    uint32 = 0xB34A // FORMAT_RRE        ADD (extended BFP)

  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,