Update ops-related pbtxt files.
PiperOrigin-RevId: 300339094 Change-Id: I9e18ed06662f6a652fd10dd623f3322e21486089
This commit is contained in:
parent
30e91ab40f
commit
ca58c8a002
19
tensorflow/core/ops/compat/ops_history_v1/BeginEpoch.pbtxt
Normal file
19
tensorflow/core/ops/compat/ops_history_v1/BeginEpoch.pbtxt
Normal file
@ -0,0 +1,19 @@
|
|||||||
|
op {
|
||||||
|
name: "BeginEpoch"
|
||||||
|
input_arg {
|
||||||
|
name: "dataset_id"
|
||||||
|
type: DT_INT64
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "address"
|
||||||
|
type: DT_STRING
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "protocol"
|
||||||
|
type: DT_STRING
|
||||||
|
}
|
||||||
|
output_arg {
|
||||||
|
name: "epoch_id"
|
||||||
|
type: DT_INT64
|
||||||
|
}
|
||||||
|
}
|
@ -0,0 +1,32 @@
|
|||||||
|
op {
|
||||||
|
name: "DataServiceDataset"
|
||||||
|
input_arg {
|
||||||
|
name: "address"
|
||||||
|
type: DT_STRING
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "protocol"
|
||||||
|
type: DT_STRING
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "max_outstanding_requests"
|
||||||
|
type: DT_INT64
|
||||||
|
}
|
||||||
|
output_arg {
|
||||||
|
name: "handle"
|
||||||
|
type: DT_VARIANT
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "output_types"
|
||||||
|
type: "list(type)"
|
||||||
|
has_minimum: true
|
||||||
|
minimum: 1
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "output_shapes"
|
||||||
|
type: "list(shape)"
|
||||||
|
has_minimum: true
|
||||||
|
minimum: 1
|
||||||
|
}
|
||||||
|
is_stateful: true
|
||||||
|
}
|
@ -0,0 +1,16 @@
|
|||||||
|
op {
|
||||||
|
name: "MakeDataServiceIterator"
|
||||||
|
input_arg {
|
||||||
|
name: "dataset"
|
||||||
|
type: DT_VARIANT
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "epoch_id"
|
||||||
|
type: DT_INT64
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "iterator"
|
||||||
|
type: DT_RESOURCE
|
||||||
|
}
|
||||||
|
is_stateful: true
|
||||||
|
}
|
@ -0,0 +1,23 @@
|
|||||||
|
op {
|
||||||
|
name: "RegisterDataset"
|
||||||
|
input_arg {
|
||||||
|
name: "dataset"
|
||||||
|
type: DT_VARIANT
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "address"
|
||||||
|
type: DT_STRING
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "protocol"
|
||||||
|
type: DT_STRING
|
||||||
|
}
|
||||||
|
output_arg {
|
||||||
|
name: "dataset_id"
|
||||||
|
type: DT_INT64
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "external_state_policy"
|
||||||
|
type: "int"
|
||||||
|
}
|
||||||
|
}
|
@ -75,3 +75,43 @@ op {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
op {
|
||||||
|
name: "ResizeArea"
|
||||||
|
input_arg {
|
||||||
|
name: "images"
|
||||||
|
type_attr: "T"
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "size"
|
||||||
|
type: DT_INT32
|
||||||
|
}
|
||||||
|
output_arg {
|
||||||
|
name: "resized_images"
|
||||||
|
type: DT_FLOAT
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "T"
|
||||||
|
type: "type"
|
||||||
|
allowed_values {
|
||||||
|
list {
|
||||||
|
type: DT_INT8
|
||||||
|
type: DT_UINT8
|
||||||
|
type: DT_INT16
|
||||||
|
type: DT_UINT16
|
||||||
|
type: DT_INT32
|
||||||
|
type: DT_INT64
|
||||||
|
type: DT_HALF
|
||||||
|
type: DT_FLOAT
|
||||||
|
type: DT_DOUBLE
|
||||||
|
type: DT_BFLOAT16
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "align_corners"
|
||||||
|
type: "bool"
|
||||||
|
default_value {
|
||||||
|
b: false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
@ -121,3 +121,50 @@ op {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
op {
|
||||||
|
name: "ResizeBicubic"
|
||||||
|
input_arg {
|
||||||
|
name: "images"
|
||||||
|
type_attr: "T"
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "size"
|
||||||
|
type: DT_INT32
|
||||||
|
}
|
||||||
|
output_arg {
|
||||||
|
name: "resized_images"
|
||||||
|
type: DT_FLOAT
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "T"
|
||||||
|
type: "type"
|
||||||
|
allowed_values {
|
||||||
|
list {
|
||||||
|
type: DT_INT8
|
||||||
|
type: DT_UINT8
|
||||||
|
type: DT_INT16
|
||||||
|
type: DT_UINT16
|
||||||
|
type: DT_INT32
|
||||||
|
type: DT_INT64
|
||||||
|
type: DT_HALF
|
||||||
|
type: DT_FLOAT
|
||||||
|
type: DT_DOUBLE
|
||||||
|
type: DT_BFLOAT16
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "align_corners"
|
||||||
|
type: "bool"
|
||||||
|
default_value {
|
||||||
|
b: false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "half_pixel_centers"
|
||||||
|
type: "bool"
|
||||||
|
default_value {
|
||||||
|
b: false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
@ -162,3 +162,51 @@ op {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
op {
|
||||||
|
name: "ResizeBilinear"
|
||||||
|
input_arg {
|
||||||
|
name: "images"
|
||||||
|
type_attr: "T"
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "size"
|
||||||
|
type: DT_INT32
|
||||||
|
}
|
||||||
|
output_arg {
|
||||||
|
name: "resized_images"
|
||||||
|
type: DT_FLOAT
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "T"
|
||||||
|
type: "type"
|
||||||
|
allowed_values {
|
||||||
|
list {
|
||||||
|
type: DT_INT8
|
||||||
|
type: DT_UINT8
|
||||||
|
type: DT_INT16
|
||||||
|
type: DT_UINT16
|
||||||
|
type: DT_INT32
|
||||||
|
type: DT_INT64
|
||||||
|
type: DT_BFLOAT16
|
||||||
|
type: DT_HALF
|
||||||
|
type: DT_FLOAT
|
||||||
|
type: DT_DOUBLE
|
||||||
|
type: DT_BFLOAT16
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "align_corners"
|
||||||
|
type: "bool"
|
||||||
|
default_value {
|
||||||
|
b: false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "half_pixel_centers"
|
||||||
|
type: "bool"
|
||||||
|
default_value {
|
||||||
|
b: false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
@ -121,3 +121,50 @@ op {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
op {
|
||||||
|
name: "ResizeNearestNeighbor"
|
||||||
|
input_arg {
|
||||||
|
name: "images"
|
||||||
|
type_attr: "T"
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "size"
|
||||||
|
type: DT_INT32
|
||||||
|
}
|
||||||
|
output_arg {
|
||||||
|
name: "resized_images"
|
||||||
|
type_attr: "T"
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "T"
|
||||||
|
type: "type"
|
||||||
|
allowed_values {
|
||||||
|
list {
|
||||||
|
type: DT_INT8
|
||||||
|
type: DT_UINT8
|
||||||
|
type: DT_INT16
|
||||||
|
type: DT_UINT16
|
||||||
|
type: DT_INT32
|
||||||
|
type: DT_INT64
|
||||||
|
type: DT_HALF
|
||||||
|
type: DT_FLOAT
|
||||||
|
type: DT_DOUBLE
|
||||||
|
type: DT_BFLOAT16
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "align_corners"
|
||||||
|
type: "bool"
|
||||||
|
default_value {
|
||||||
|
b: false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "half_pixel_centers"
|
||||||
|
type: "bool"
|
||||||
|
default_value {
|
||||||
|
b: false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
@ -28,3 +28,41 @@ op {
|
|||||||
}
|
}
|
||||||
is_stateful: true
|
is_stateful: true
|
||||||
}
|
}
|
||||||
|
op {
|
||||||
|
name: "VarHandleOp"
|
||||||
|
output_arg {
|
||||||
|
name: "resource"
|
||||||
|
type: DT_RESOURCE
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "container"
|
||||||
|
type: "string"
|
||||||
|
default_value {
|
||||||
|
s: ""
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "shared_name"
|
||||||
|
type: "string"
|
||||||
|
default_value {
|
||||||
|
s: ""
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "dtype"
|
||||||
|
type: "type"
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "shape"
|
||||||
|
type: "shape"
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "allowed_devices"
|
||||||
|
type: "list(string)"
|
||||||
|
default_value {
|
||||||
|
list {
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
is_stateful: true
|
||||||
|
}
|
||||||
|
@ -4256,6 +4256,25 @@ op {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
op {
|
||||||
|
name: "BeginEpoch"
|
||||||
|
input_arg {
|
||||||
|
name: "dataset_id"
|
||||||
|
type: DT_INT64
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "address"
|
||||||
|
type: DT_STRING
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "protocol"
|
||||||
|
type: DT_STRING
|
||||||
|
}
|
||||||
|
output_arg {
|
||||||
|
name: "epoch_id"
|
||||||
|
type: DT_INT64
|
||||||
|
}
|
||||||
|
}
|
||||||
op {
|
op {
|
||||||
name: "BesselI0e"
|
name: "BesselI0e"
|
||||||
input_arg {
|
input_arg {
|
||||||
@ -10435,6 +10454,38 @@ op {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
op {
|
||||||
|
name: "DataServiceDataset"
|
||||||
|
input_arg {
|
||||||
|
name: "address"
|
||||||
|
type: DT_STRING
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "protocol"
|
||||||
|
type: DT_STRING
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "max_outstanding_requests"
|
||||||
|
type: DT_INT64
|
||||||
|
}
|
||||||
|
output_arg {
|
||||||
|
name: "handle"
|
||||||
|
type: DT_VARIANT
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "output_types"
|
||||||
|
type: "list(type)"
|
||||||
|
has_minimum: true
|
||||||
|
minimum: 1
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "output_shapes"
|
||||||
|
type: "list(shape)"
|
||||||
|
has_minimum: true
|
||||||
|
minimum: 1
|
||||||
|
}
|
||||||
|
is_stateful: true
|
||||||
|
}
|
||||||
op {
|
op {
|
||||||
name: "DatasetCardinality"
|
name: "DatasetCardinality"
|
||||||
input_arg {
|
input_arg {
|
||||||
@ -21275,6 +21326,22 @@ op {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
op {
|
||||||
|
name: "MakeDataServiceIterator"
|
||||||
|
input_arg {
|
||||||
|
name: "dataset"
|
||||||
|
type: DT_VARIANT
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "epoch_id"
|
||||||
|
type: DT_INT64
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "iterator"
|
||||||
|
type: DT_RESOURCE
|
||||||
|
}
|
||||||
|
is_stateful: true
|
||||||
|
}
|
||||||
op {
|
op {
|
||||||
name: "MakeIterator"
|
name: "MakeIterator"
|
||||||
input_arg {
|
input_arg {
|
||||||
@ -34571,6 +34638,29 @@ op {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
op {
|
||||||
|
name: "RegisterDataset"
|
||||||
|
input_arg {
|
||||||
|
name: "dataset"
|
||||||
|
type: DT_VARIANT
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "address"
|
||||||
|
type: DT_STRING
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "protocol"
|
||||||
|
type: DT_STRING
|
||||||
|
}
|
||||||
|
output_arg {
|
||||||
|
name: "dataset_id"
|
||||||
|
type: DT_INT64
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "external_state_policy"
|
||||||
|
type: "int"
|
||||||
|
}
|
||||||
|
}
|
||||||
op {
|
op {
|
||||||
name: "Relu"
|
name: "Relu"
|
||||||
input_arg {
|
input_arg {
|
||||||
@ -35055,6 +35145,7 @@ op {
|
|||||||
type: DT_HALF
|
type: DT_HALF
|
||||||
type: DT_FLOAT
|
type: DT_FLOAT
|
||||||
type: DT_DOUBLE
|
type: DT_DOUBLE
|
||||||
|
type: DT_BFLOAT16
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -35094,6 +35185,7 @@ op {
|
|||||||
type: DT_HALF
|
type: DT_HALF
|
||||||
type: DT_FLOAT
|
type: DT_FLOAT
|
||||||
type: DT_DOUBLE
|
type: DT_DOUBLE
|
||||||
|
type: DT_BFLOAT16
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -35180,6 +35272,7 @@ op {
|
|||||||
type: DT_HALF
|
type: DT_HALF
|
||||||
type: DT_FLOAT
|
type: DT_FLOAT
|
||||||
type: DT_DOUBLE
|
type: DT_DOUBLE
|
||||||
|
type: DT_BFLOAT16
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -35267,6 +35360,7 @@ op {
|
|||||||
type: DT_HALF
|
type: DT_HALF
|
||||||
type: DT_FLOAT
|
type: DT_FLOAT
|
||||||
type: DT_DOUBLE
|
type: DT_DOUBLE
|
||||||
|
type: DT_BFLOAT16
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -52381,6 +52475,14 @@ op {
|
|||||||
name: "shape"
|
name: "shape"
|
||||||
type: "shape"
|
type: "shape"
|
||||||
}
|
}
|
||||||
|
attr {
|
||||||
|
name: "allowed_devices"
|
||||||
|
type: "list(string)"
|
||||||
|
default_value {
|
||||||
|
list {
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
is_stateful: true
|
is_stateful: true
|
||||||
}
|
}
|
||||||
op {
|
op {
|
||||||
|
Loading…
Reference in New Issue
Block a user