is_stateful: true
}
op {
+ name: "StatelessMultinomial"
+ input_arg {
+ name: "logits"
+ type_attr: "T"
+ }
+ input_arg {
+ name: "num_samples"
+ type: DT_INT32
+ }
+ input_arg {
+ name: "seed"
+ type_attr: "Tseed"
+ }
+ output_arg {
+ name: "output"
+ type_attr: "output_dtype"
+ }
+ attr {
+ name: "T"
+ type: "type"
+ allowed_values {
+ list {
+ type: DT_FLOAT
+ type: DT_DOUBLE
+ type: DT_INT32
+ type: DT_UINT8
+ type: DT_INT16
+ type: DT_INT8
+ type: DT_INT64
+ type: DT_BFLOAT16
+ type: DT_UINT16
+ type: DT_HALF
+ type: DT_UINT32
+ type: DT_UINT64
+ }
+ }
+ }
+ attr {
+ name: "Tseed"
+ type: "type"
+ default_value {
+ type: DT_INT64
+ }
+ allowed_values {
+ list {
+ type: DT_INT32
+ type: DT_INT64
+ }
+ }
+ }
+ attr {
+ name: "output_dtype"
+ type: "type"
+ default_value {
+ type: DT_INT64
+ }
+ allowed_values {
+ list {
+ type: DT_INT32
+ type: DT_INT64
+ }
+ }
+ }
+}
+op {
name: "StatelessRandomNormal"
input_arg {
name: "shape"
is_stateful: true
}
op {
+ name: "StatelessMultinomial"
+ input_arg {
+ name: "logits"
+ type_attr: "T"
+ }
+ input_arg {
+ name: "num_samples"
+ type: DT_INT32
+ }
+ input_arg {
+ name: "seed"
+ type_attr: "Tseed"
+ }
+ output_arg {
+ name: "output"
+ type_attr: "output_dtype"
+ }
+ attr {
+ name: "T"
+ type: "type"
+ allowed_values {
+ list {
+ type: DT_FLOAT
+ type: DT_DOUBLE
+ type: DT_INT32
+ type: DT_UINT8
+ type: DT_INT16
+ type: DT_INT8
+ type: DT_INT64
+ type: DT_BFLOAT16
+ type: DT_UINT16
+ type: DT_HALF
+ type: DT_UINT32
+ type: DT_UINT64
+ }
+ }
+ }
+ attr {
+ name: "Tseed"
+ type: "type"
+ default_value {
+ type: DT_INT64
+ }
+ allowed_values {
+ list {
+ type: DT_INT32
+ type: DT_INT64
+ }
+ }
+ }
+ attr {
+ name: "output_dtype"
+ type: "type"
+ default_value {
+ type: DT_INT64
+ }
+ allowed_values {
+ list {
+ type: DT_INT32
+ type: DT_INT64
+ }
+ }
+ }
+}
+op {
name: "StatelessRandomNormal"
input_arg {
name: "shape"