Update ops-related pbtxt files.
PiperOrigin-RevId: 323515788 Change-Id: I5f1f509c2608be714a5951bccd447b73bd1b77e9
This commit is contained in:
parent
c8dcee2b65
commit
e8029af3b2
@ -0,0 +1,43 @@
|
||||
op {
|
||||
name: "OptimizeDatasetV2"
|
||||
input_arg {
|
||||
name: "input_dataset"
|
||||
type: DT_VARIANT
|
||||
}
|
||||
input_arg {
|
||||
name: "optimizations_enabled"
|
||||
type: DT_STRING
|
||||
}
|
||||
input_arg {
|
||||
name: "optimizations_disabled"
|
||||
type: DT_STRING
|
||||
}
|
||||
input_arg {
|
||||
name: "optimizations_default"
|
||||
type: DT_STRING
|
||||
}
|
||||
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
|
||||
}
|
||||
attr {
|
||||
name: "optimization_configs"
|
||||
type: "list(string)"
|
||||
default_value {
|
||||
list {
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
@ -26317,6 +26317,49 @@ op {
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "OptimizeDatasetV2"
|
||||
input_arg {
|
||||
name: "input_dataset"
|
||||
type: DT_VARIANT
|
||||
}
|
||||
input_arg {
|
||||
name: "optimizations_enabled"
|
||||
type: DT_STRING
|
||||
}
|
||||
input_arg {
|
||||
name: "optimizations_disabled"
|
||||
type: DT_STRING
|
||||
}
|
||||
input_arg {
|
||||
name: "optimizations_default"
|
||||
type: DT_STRING
|
||||
}
|
||||
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
|
||||
}
|
||||
attr {
|
||||
name: "optimization_configs"
|
||||
type: "list(string)"
|
||||
default_value {
|
||||
list {
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "OptionalFromValue"
|
||||
input_arg {
|
||||
|
Loading…
x
Reference in New Issue
Block a user