Update ops-related pbtxt files.

PiperOrigin-RevId: 353341405
Change-Id: Ied9772e1b2fbc7a92c0118e0f6eac54c1618b862
This commit is contained in:
A. Unique TensorFlower 2021-01-22 16:29:24 -08:00 committed by TensorFlower Gardener
parent 8bf282a345
commit 8d33fd7426
17 changed files with 840 additions and 0 deletions

View File

@ -0,0 +1,23 @@
op {
name: "RiscAbs"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}

View File

@ -0,0 +1,23 @@
op {
name: "RiscCeil"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}

View File

@ -0,0 +1,23 @@
op {
name: "RiscCos"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}

View File

@ -0,0 +1,27 @@
op {
name: "RiscDiv"
input_arg {
name: "x"
type_attr: "T"
}
input_arg {
name: "y"
type_attr: "T"
}
output_arg {
name: "z"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}

View File

@ -0,0 +1,23 @@
op {
name: "RiscExp"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}

View File

@ -0,0 +1,23 @@
op {
name: "RiscFloor"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}

View File

@ -0,0 +1,37 @@
op {
name: "RiscImag"
input_arg {
name: "input"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "Tout"
}
attr {
name: "T"
type: "type"
default_value {
type: DT_COMPLEX64
}
allowed_values {
list {
type: DT_COMPLEX64
type: DT_COMPLEX128
}
}
}
attr {
name: "Tout"
type: "type"
default_value {
type: DT_FLOAT
}
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}

View File

@ -0,0 +1,23 @@
op {
name: "RiscLog"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}

View File

@ -0,0 +1,27 @@
op {
name: "RiscMin"
input_arg {
name: "x"
type_attr: "T"
}
input_arg {
name: "y"
type_attr: "T"
}
output_arg {
name: "z"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}

View File

@ -0,0 +1,27 @@
op {
name: "RiscMul"
input_arg {
name: "x"
type_attr: "T"
}
input_arg {
name: "y"
type_attr: "T"
}
output_arg {
name: "z"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}

View File

@ -0,0 +1,23 @@
op {
name: "RiscNeg"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}

View File

@ -0,0 +1,27 @@
op {
name: "RiscPow"
input_arg {
name: "x"
type_attr: "T"
}
input_arg {
name: "y"
type_attr: "T"
}
output_arg {
name: "z"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}

View File

@ -0,0 +1,37 @@
op {
name: "RiscReal"
input_arg {
name: "input"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "Tout"
}
attr {
name: "T"
type: "type"
default_value {
type: DT_COMPLEX64
}
allowed_values {
list {
type: DT_COMPLEX64
type: DT_COMPLEX128
}
}
}
attr {
name: "Tout"
type: "type"
default_value {
type: DT_FLOAT
}
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}

View File

@ -0,0 +1,27 @@
op {
name: "RiscRem"
input_arg {
name: "x"
type_attr: "T"
}
input_arg {
name: "y"
type_attr: "T"
}
output_arg {
name: "z"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}

View File

@ -0,0 +1,23 @@
op {
name: "RiscSign"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}

View File

@ -0,0 +1,27 @@
op {
name: "RiscSub"
input_arg {
name: "x"
type_attr: "T"
}
input_arg {
name: "y"
type_attr: "T"
}
output_arg {
name: "z"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}

View File

@ -41683,6 +41683,29 @@ op {
}
}
}
op {
name: "RiscAbs"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}
op {
name: "RiscAdd"
input_arg {
@ -41865,6 +41888,29 @@ op {
type: "type"
}
}
op {
name: "RiscCeil"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}
op {
name: "RiscCholesky"
input_arg {
@ -42034,6 +42080,56 @@ op {
}
}
}
op {
name: "RiscCos"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}
op {
name: "RiscDiv"
input_arg {
name: "x"
type_attr: "T"
}
input_arg {
name: "y"
type_attr: "T"
}
output_arg {
name: "z"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}
op {
name: "RiscDot"
input_arg {
@ -42075,6 +42171,29 @@ op {
}
}
}
op {
name: "RiscExp"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}
op {
name: "RiscFft"
input_arg {
@ -42099,6 +42218,29 @@ op {
}
}
}
op {
name: "RiscFloor"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}
op {
name: "RiscGather"
input_arg {
@ -42149,6 +42291,43 @@ op {
}
}
}
op {
name: "RiscImag"
input_arg {
name: "input"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "Tout"
}
attr {
name: "T"
type: "type"
default_value {
type: DT_COMPLEX64
}
allowed_values {
list {
type: DT_COMPLEX64
type: DT_COMPLEX128
}
}
}
attr {
name: "Tout"
type: "type"
default_value {
type: DT_FLOAT
}
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}
op {
name: "RiscIsFinite"
input_arg {
@ -42172,6 +42351,29 @@ op {
}
}
}
op {
name: "RiscLog"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}
op {
name: "RiscLogicalAnd"
input_arg {
@ -42240,6 +42442,83 @@ op {
}
}
}
op {
name: "RiscMin"
input_arg {
name: "x"
type_attr: "T"
}
input_arg {
name: "y"
type_attr: "T"
}
output_arg {
name: "z"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}
op {
name: "RiscMul"
input_arg {
name: "x"
type_attr: "T"
}
input_arg {
name: "y"
type_attr: "T"
}
output_arg {
name: "z"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}
op {
name: "RiscNeg"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}
op {
name: "RiscPad"
input_arg {
@ -42342,6 +42621,33 @@ op {
}
}
}
op {
name: "RiscPow"
input_arg {
name: "x"
type_attr: "T"
}
input_arg {
name: "y"
type_attr: "T"
}
output_arg {
name: "z"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}
op {
name: "RiscRandomUniform"
input_arg {
@ -42370,6 +42676,43 @@ op {
}
}
}
op {
name: "RiscReal"
input_arg {
name: "input"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "Tout"
}
attr {
name: "T"
type: "type"
default_value {
type: DT_COMPLEX64
}
allowed_values {
list {
type: DT_COMPLEX64
type: DT_COMPLEX128
}
}
}
attr {
name: "Tout"
type: "type"
default_value {
type: DT_FLOAT
}
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}
op {
name: "RiscReduce"
input_arg {
@ -42420,6 +42763,33 @@ op {
}
}
}
op {
name: "RiscRem"
input_arg {
name: "x"
type_attr: "T"
}
input_arg {
name: "y"
type_attr: "T"
}
output_arg {
name: "z"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}
op {
name: "RiscReshape"
input_arg {
@ -42577,6 +42947,29 @@ op {
}
}
}
op {
name: "RiscSign"
input_arg {
name: "x"
type_attr: "T"
}
output_arg {
name: "y"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}
op {
name: "RiscSlice"
input_arg {
@ -42692,6 +43085,33 @@ op {
has_minimum: true
}
}
op {
name: "RiscSub"
input_arg {
name: "x"
type_attr: "T"
}
input_arg {
name: "y"
type_attr: "T"
}
output_arg {
name: "z"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_BFLOAT16
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
}
op {
name: "RiscTranspose"
input_arg {