From 2177c5597b60829c45f40b2aede729d858390067 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Mon, 10 Jun 2019 23:18:46 -0700 Subject: [PATCH] Update ops-related pbtxt files. PiperOrigin-RevId: 252556834 --- .../core/ops/compat/ops_history.v1.pbtxt | 653 ++++++++++++++++++ tensorflow/core/ops/ops.pbtxt | 265 +++++++ 2 files changed, 918 insertions(+) diff --git a/tensorflow/core/ops/compat/ops_history.v1.pbtxt b/tensorflow/core/ops/compat/ops_history.v1.pbtxt index bd7e4b038a1..0a5b095f8f3 100644 --- a/tensorflow/core/ops/compat/ops_history.v1.pbtxt +++ b/tensorflow/core/ops/compat/ops_history.v1.pbtxt @@ -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 { diff --git a/tensorflow/core/ops/ops.pbtxt b/tensorflow/core/ops/ops.pbtxt index cbdc4bafab0..d601712bec7 100644 --- a/tensorflow/core/ops/ops.pbtxt +++ b/tensorflow/core/ops/ops.pbtxt @@ -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"