message: Adds double support for StatelessMultinomial (which is really a categorical).
  Relevant to
diff: @@ -186,7 +186,7 @@ class StatelessCategoricalOp : public CategoricalOp {
  REGISTER_XLA_OP(Name("StatelessMultinomial")
  .CompileTimeConstantInput("num_samples")
  - .TypeConstraint("T", {DT_FLOAT, DT_BFLOAT16})
  + .TypeConstraint("T", {DT_DOUBLE, DT_FLOAT, DT_BFLOAT16})
  .TypeConstraint("Tseed", DT_INT32),
  StatelessCategoricalOp);
  