Update ops-related pbtxt files.
PiperOrigin-RevId: 252556834
This commit is contained in:
parent
9380a41290
commit
2177c5597b
@ -17919,6 +17919,166 @@ op {
|
||||
}
|
||||
is_stateful: true
|
||||
}
|
||||
op {
|
||||
name: "CudnnRNNBackpropV3"
|
||||
input_arg {
|
||||
name: "input"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "input_h"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "input_c"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "params"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "sequence_lengths"
|
||||
type: DT_INT32
|
||||
}
|
||||
input_arg {
|
||||
name: "output"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "output_h"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "output_c"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "output_backprop"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "output_h_backprop"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "output_c_backprop"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "reserve_space"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "host_reserved"
|
||||
type: DT_INT8
|
||||
}
|
||||
output_arg {
|
||||
name: "input_backprop"
|
||||
type_attr: "T"
|
||||
}
|
||||
output_arg {
|
||||
name: "input_h_backprop"
|
||||
type_attr: "T"
|
||||
}
|
||||
output_arg {
|
||||
name: "input_c_backprop"
|
||||
type_attr: "T"
|
||||
}
|
||||
output_arg {
|
||||
name: "params_backprop"
|
||||
type_attr: "T"
|
||||
}
|
||||
attr {
|
||||
name: "T"
|
||||
type: "type"
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_HALF
|
||||
type: DT_FLOAT
|
||||
type: DT_DOUBLE
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "rnn_mode"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: "lstm"
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
s: "rnn_relu"
|
||||
s: "rnn_tanh"
|
||||
s: "lstm"
|
||||
s: "gru"
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "input_mode"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: "linear_input"
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
s: "linear_input"
|
||||
s: "skip_input"
|
||||
s: "auto_select"
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "direction"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: "unidirectional"
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
s: "unidirectional"
|
||||
s: "bidirectional"
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "dropout"
|
||||
type: "float"
|
||||
default_value {
|
||||
f: 0
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "seed"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: 0
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "seed2"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: 0
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "num_proj"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: 0
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "time_major"
|
||||
type: "bool"
|
||||
default_value {
|
||||
b: true
|
||||
}
|
||||
}
|
||||
is_stateful: true
|
||||
}
|
||||
op {
|
||||
name: "CudnnRNNCanonicalToParams"
|
||||
input_arg {
|
||||
@ -18028,6 +18188,231 @@ op {
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "CudnnRNNCanonicalToParamsV2"
|
||||
input_arg {
|
||||
name: "num_layers"
|
||||
type: DT_INT32
|
||||
}
|
||||
input_arg {
|
||||
name: "num_units"
|
||||
type: DT_INT32
|
||||
}
|
||||
input_arg {
|
||||
name: "input_size"
|
||||
type: DT_INT32
|
||||
}
|
||||
input_arg {
|
||||
name: "weights"
|
||||
type_attr: "T"
|
||||
number_attr: "num_params_weights"
|
||||
}
|
||||
input_arg {
|
||||
name: "biases"
|
||||
type_attr: "T"
|
||||
number_attr: "num_params_biases"
|
||||
}
|
||||
output_arg {
|
||||
name: "params"
|
||||
type_attr: "T"
|
||||
}
|
||||
attr {
|
||||
name: "T"
|
||||
type: "type"
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_HALF
|
||||
type: DT_FLOAT
|
||||
type: DT_DOUBLE
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "num_params_weights"
|
||||
type: "int"
|
||||
has_minimum: true
|
||||
minimum: 1
|
||||
}
|
||||
attr {
|
||||
name: "num_params_biases"
|
||||
type: "int"
|
||||
has_minimum: true
|
||||
minimum: 1
|
||||
}
|
||||
attr {
|
||||
name: "rnn_mode"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: "lstm"
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
s: "rnn_relu"
|
||||
s: "rnn_tanh"
|
||||
s: "lstm"
|
||||
s: "gru"
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "input_mode"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: "linear_input"
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
s: "linear_input"
|
||||
s: "skip_input"
|
||||
s: "auto_select"
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "direction"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: "unidirectional"
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
s: "unidirectional"
|
||||
s: "bidirectional"
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "dropout"
|
||||
type: "float"
|
||||
default_value {
|
||||
f: 0
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "seed"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: 0
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "seed2"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: 0
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "num_proj"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: 0
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "CudnnRNNParamsSize"
|
||||
input_arg {
|
||||
name: "num_layers"
|
||||
type: DT_INT32
|
||||
}
|
||||
input_arg {
|
||||
name: "num_units"
|
||||
type: DT_INT32
|
||||
}
|
||||
input_arg {
|
||||
name: "input_size"
|
||||
type: DT_INT32
|
||||
}
|
||||
output_arg {
|
||||
name: "params_size"
|
||||
type_attr: "S"
|
||||
}
|
||||
attr {
|
||||
name: "T"
|
||||
type: "type"
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_HALF
|
||||
type: DT_FLOAT
|
||||
type: DT_DOUBLE
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "S"
|
||||
type: "type"
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_INT32
|
||||
type: DT_INT64
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "rnn_mode"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: "lstm"
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
s: "rnn_relu"
|
||||
s: "rnn_tanh"
|
||||
s: "lstm"
|
||||
s: "gru"
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "input_mode"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: "linear_input"
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
s: "linear_input"
|
||||
s: "skip_input"
|
||||
s: "auto_select"
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "direction"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: "unidirectional"
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
s: "unidirectional"
|
||||
s: "bidirectional"
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "dropout"
|
||||
type: "float"
|
||||
default_value {
|
||||
f: 0
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "seed"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: 0
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "seed2"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: 0
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "CudnnRNNParamsSize"
|
||||
input_arg {
|
||||
@ -18130,6 +18515,13 @@ op {
|
||||
i: 0
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "num_proj"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: 0
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "CudnnRNNParamsToCanonical"
|
||||
@ -18240,6 +18632,128 @@ op {
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "CudnnRNNParamsToCanonicalV2"
|
||||
input_arg {
|
||||
name: "num_layers"
|
||||
type: DT_INT32
|
||||
}
|
||||
input_arg {
|
||||
name: "num_units"
|
||||
type: DT_INT32
|
||||
}
|
||||
input_arg {
|
||||
name: "input_size"
|
||||
type: DT_INT32
|
||||
}
|
||||
input_arg {
|
||||
name: "params"
|
||||
type_attr: "T"
|
||||
}
|
||||
output_arg {
|
||||
name: "weights"
|
||||
type_attr: "T"
|
||||
number_attr: "num_params_weights"
|
||||
}
|
||||
output_arg {
|
||||
name: "biases"
|
||||
type_attr: "T"
|
||||
number_attr: "num_params_biases"
|
||||
}
|
||||
attr {
|
||||
name: "T"
|
||||
type: "type"
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_HALF
|
||||
type: DT_FLOAT
|
||||
type: DT_DOUBLE
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "num_params_weights"
|
||||
type: "int"
|
||||
has_minimum: true
|
||||
minimum: 1
|
||||
}
|
||||
attr {
|
||||
name: "num_params_biases"
|
||||
type: "int"
|
||||
has_minimum: true
|
||||
minimum: 1
|
||||
}
|
||||
attr {
|
||||
name: "rnn_mode"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: "lstm"
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
s: "rnn_relu"
|
||||
s: "rnn_tanh"
|
||||
s: "lstm"
|
||||
s: "gru"
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "input_mode"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: "linear_input"
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
s: "linear_input"
|
||||
s: "skip_input"
|
||||
s: "auto_select"
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "direction"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: "unidirectional"
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
s: "unidirectional"
|
||||
s: "bidirectional"
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "dropout"
|
||||
type: "float"
|
||||
default_value {
|
||||
f: 0
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "seed"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: 0
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "seed2"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: 0
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "num_proj"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: 0
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "CudnnRNNV2"
|
||||
input_arg {
|
||||
@ -18618,6 +19132,145 @@ op {
|
||||
}
|
||||
is_stateful: true
|
||||
}
|
||||
op {
|
||||
name: "CudnnRNNV3"
|
||||
input_arg {
|
||||
name: "input"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "input_h"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "input_c"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "params"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "sequence_lengths"
|
||||
type: DT_INT32
|
||||
}
|
||||
output_arg {
|
||||
name: "output"
|
||||
type_attr: "T"
|
||||
}
|
||||
output_arg {
|
||||
name: "output_h"
|
||||
type_attr: "T"
|
||||
}
|
||||
output_arg {
|
||||
name: "output_c"
|
||||
type_attr: "T"
|
||||
}
|
||||
output_arg {
|
||||
name: "reserve_space"
|
||||
type_attr: "T"
|
||||
}
|
||||
output_arg {
|
||||
name: "host_reserved"
|
||||
type: DT_INT8
|
||||
}
|
||||
attr {
|
||||
name: "T"
|
||||
type: "type"
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_HALF
|
||||
type: DT_FLOAT
|
||||
type: DT_DOUBLE
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "rnn_mode"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: "lstm"
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
s: "rnn_relu"
|
||||
s: "rnn_tanh"
|
||||
s: "lstm"
|
||||
s: "gru"
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "input_mode"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: "linear_input"
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
s: "linear_input"
|
||||
s: "skip_input"
|
||||
s: "auto_select"
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "direction"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: "unidirectional"
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
s: "unidirectional"
|
||||
s: "bidirectional"
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "dropout"
|
||||
type: "float"
|
||||
default_value {
|
||||
f: 0
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "seed"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: 0
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "seed2"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: 0
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "num_proj"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: 0
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "is_training"
|
||||
type: "bool"
|
||||
default_value {
|
||||
b: true
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "time_major"
|
||||
type: "bool"
|
||||
default_value {
|
||||
b: true
|
||||
}
|
||||
}
|
||||
is_stateful: true
|
||||
}
|
||||
op {
|
||||
name: "Cumprod"
|
||||
input_arg {
|
||||
|
@ -8221,6 +8221,13 @@ op {
|
||||
i: 0
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "num_proj"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: 0
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "time_major"
|
||||
type: "bool"
|
||||
@ -8339,6 +8346,128 @@ op {
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "CudnnRNNCanonicalToParamsV2"
|
||||
input_arg {
|
||||
name: "num_layers"
|
||||
type: DT_INT32
|
||||
}
|
||||
input_arg {
|
||||
name: "num_units"
|
||||
type: DT_INT32
|
||||
}
|
||||
input_arg {
|
||||
name: "input_size"
|
||||
type: DT_INT32
|
||||
}
|
||||
input_arg {
|
||||
name: "weights"
|
||||
type_attr: "T"
|
||||
number_attr: "num_params_weights"
|
||||
}
|
||||
input_arg {
|
||||
name: "biases"
|
||||
type_attr: "T"
|
||||
number_attr: "num_params_biases"
|
||||
}
|
||||
output_arg {
|
||||
name: "params"
|
||||
type_attr: "T"
|
||||
}
|
||||
attr {
|
||||
name: "T"
|
||||
type: "type"
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_HALF
|
||||
type: DT_FLOAT
|
||||
type: DT_DOUBLE
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "num_params_weights"
|
||||
type: "int"
|
||||
has_minimum: true
|
||||
minimum: 1
|
||||
}
|
||||
attr {
|
||||
name: "num_params_biases"
|
||||
type: "int"
|
||||
has_minimum: true
|
||||
minimum: 1
|
||||
}
|
||||
attr {
|
||||
name: "rnn_mode"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: "lstm"
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
s: "rnn_relu"
|
||||
s: "rnn_tanh"
|
||||
s: "lstm"
|
||||
s: "gru"
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "input_mode"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: "linear_input"
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
s: "linear_input"
|
||||
s: "skip_input"
|
||||
s: "auto_select"
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "direction"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: "unidirectional"
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
s: "unidirectional"
|
||||
s: "bidirectional"
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "dropout"
|
||||
type: "float"
|
||||
default_value {
|
||||
f: 0
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "seed"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: 0
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "seed2"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: 0
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "num_proj"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: 0
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "CudnnRNNParamsSize"
|
||||
input_arg {
|
||||
@ -8441,6 +8570,13 @@ op {
|
||||
i: 0
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "num_proj"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: 0
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "CudnnRNNParamsToCanonical"
|
||||
@ -8551,6 +8687,128 @@ op {
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "CudnnRNNParamsToCanonicalV2"
|
||||
input_arg {
|
||||
name: "num_layers"
|
||||
type: DT_INT32
|
||||
}
|
||||
input_arg {
|
||||
name: "num_units"
|
||||
type: DT_INT32
|
||||
}
|
||||
input_arg {
|
||||
name: "input_size"
|
||||
type: DT_INT32
|
||||
}
|
||||
input_arg {
|
||||
name: "params"
|
||||
type_attr: "T"
|
||||
}
|
||||
output_arg {
|
||||
name: "weights"
|
||||
type_attr: "T"
|
||||
number_attr: "num_params_weights"
|
||||
}
|
||||
output_arg {
|
||||
name: "biases"
|
||||
type_attr: "T"
|
||||
number_attr: "num_params_biases"
|
||||
}
|
||||
attr {
|
||||
name: "T"
|
||||
type: "type"
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_HALF
|
||||
type: DT_FLOAT
|
||||
type: DT_DOUBLE
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "num_params_weights"
|
||||
type: "int"
|
||||
has_minimum: true
|
||||
minimum: 1
|
||||
}
|
||||
attr {
|
||||
name: "num_params_biases"
|
||||
type: "int"
|
||||
has_minimum: true
|
||||
minimum: 1
|
||||
}
|
||||
attr {
|
||||
name: "rnn_mode"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: "lstm"
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
s: "rnn_relu"
|
||||
s: "rnn_tanh"
|
||||
s: "lstm"
|
||||
s: "gru"
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "input_mode"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: "linear_input"
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
s: "linear_input"
|
||||
s: "skip_input"
|
||||
s: "auto_select"
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "direction"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: "unidirectional"
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
s: "unidirectional"
|
||||
s: "bidirectional"
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "dropout"
|
||||
type: "float"
|
||||
default_value {
|
||||
f: 0
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "seed"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: 0
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "seed2"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: 0
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "num_proj"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: 0
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "CudnnRNNV2"
|
||||
input_arg {
|
||||
@ -8788,6 +9046,13 @@ op {
|
||||
i: 0
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "num_proj"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: 0
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "is_training"
|
||||
type: "bool"
|
||||
|
Loading…
x
Reference in New Issue
Block a user