Update ops-related pbtxt files.

PiperOrigin-RevId: 237542628
This commit is contained in:
A. Unique TensorFlower 2019-03-08 16:37:49 -08:00 committed by TensorFlower Gardener
parent c704ff279d
commit 7d52b0c2d2
2 changed files with 299 additions and 0 deletions

View File

@ -17166,6 +17166,159 @@ 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: "time_major"
type: "bool"
default_value {
b: true
}
}
is_stateful: true
}
op {
name: "CudnnRNNCanonicalToParams"
input_arg {
@ -17733,6 +17886,138 @@ 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: "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 {

View File

@ -7702,6 +7702,13 @@ op {
i: 0
}
}
attr {
name: "time_major"
type: "bool"
default_value {
b: true
}
}
is_stateful: true
}
op {
@ -8269,6 +8276,13 @@ op {
b: true
}
}
attr {
name: "time_major"
type: "bool"
default_value {
b: true
}
}
is_stateful: true
}
op {