diff options
-rw-r--r-- | tensorflow/core/ops/compat/ops_history.v0.pbtxt | 47 | ||||
-rw-r--r-- | tensorflow/core/ops/ops.pbtxt | 1 |
2 files changed, 48 insertions, 0 deletions
diff --git a/tensorflow/core/ops/compat/ops_history.v0.pbtxt b/tensorflow/core/ops/compat/ops_history.v0.pbtxt index df351827f5..4e93d0d449 100644 --- a/tensorflow/core/ops/compat/ops_history.v0.pbtxt +++ b/tensorflow/core/ops/compat/ops_history.v0.pbtxt @@ -11605,6 +11605,53 @@ op { } } op { + name: "Multinomial" + input_arg { + name: "logits" + type_attr: "T" + } + input_arg { + name: "num_samples" + type: DT_INT32 + } + output_arg { + name: "output" + type: DT_INT64 + } + attr { + name: "seed" + type: "int" + default_value { + i: 0 + } + } + attr { + name: "seed2" + type: "int" + default_value { + i: 0 + } + } + attr { + name: "T" + type: "type" + allowed_values { + list { + type: DT_FLOAT + type: DT_DOUBLE + type: DT_INT32 + type: DT_INT64 + type: DT_UINT8 + type: DT_INT16 + type: DT_INT8 + type: DT_UINT16 + type: DT_HALF + } + } + } + is_stateful: true +} +op { name: "Neg" input_arg { name: "x" diff --git a/tensorflow/core/ops/ops.pbtxt b/tensorflow/core/ops/ops.pbtxt index d585101a5f..f3be56f9b8 100644 --- a/tensorflow/core/ops/ops.pbtxt +++ b/tensorflow/core/ops/ops.pbtxt @@ -6903,6 +6903,7 @@ op { } } summary: "Draws samples from a multinomial distribution." + is_stateful: true } op { name: "Neg" |