Update ops-related pbtxt files.

PiperOrigin-RevId: 300339094
Change-Id: I9e18ed06662f6a652fd10dd623f3322e21486089
This commit is contained in:
A. Unique TensorFlower 2020-03-11 09:17:51 -07:00 committed by TensorFlower Gardener
parent 30e91ab40f
commit ca58c8a002
10 changed files with 412 additions and 0 deletions

View 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
}
}

View File

@ -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
}

View File

@ -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
}

View File

@ -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"
}
}

View File

@ -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
}
}
}

View File

@ -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
}
}
}

View File

@ -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
}
}
}

View File

@ -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
}
}
}

View File

@ -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
}

View File

@ -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 {