blob: ce8a7f8e20f6ef41843f388bd1dd746fddef1367 [file] [log] [blame]
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
}
}
}