| 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 |
| } |
| } |
| 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: "time_major" |
| type: "bool" |
| default_value { |
| b: true |
| } |
| } |
| 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 |
| } |