Update ops-related pbtxt files.

PiperOrigin-RevId: 351448163
Change-Id: I1a499506395d81dcb752c2018e1ddd3666fe61d8
This commit is contained in:
A. Unique TensorFlower 2021-01-12 14:18:33 -08:00 committed by TensorFlower Gardener
parent 5edd67fd01
commit 0ead9dec60
3 changed files with 39 additions and 0 deletions

View File

@ -5,3 +5,11 @@ op {
type: DT_INT32
}
}
op {
name: "StatelessRandomGetAlg"
output_arg {
name: "alg"
type: DT_INT32
}
is_stateful: true
}

View File

@ -26,3 +26,32 @@ op {
}
}
}
op {
name: "StatelessRandomGetKeyCounter"
input_arg {
name: "seed"
type_attr: "Tseed"
}
output_arg {
name: "key"
type: DT_UINT64
}
output_arg {
name: "counter"
type: DT_UINT64
}
attr {
name: "Tseed"
type: "type"
default_value {
type: DT_INT64
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
is_stateful: true
}

View File

@ -51346,6 +51346,7 @@ op {
name: "alg"
type: DT_INT32
}
is_stateful: true
}
op {
name: "StatelessRandomGetKeyCounter"
@ -51374,6 +51375,7 @@ op {
}
}
}
is_stateful: true
}
op {
name: "StatelessRandomGetKeyCounterAlg"