Update ops-related pbtxt files.
PiperOrigin-RevId: 313289048 Change-Id: I7fd33d96318f8f0c03b538defbba16eb1044c38b
This commit is contained in:
parent
3f423f882b
commit
ba07462168
@ -56,3 +56,46 @@ op {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
op {
|
||||||
|
name: "TPUReplicatedInput"
|
||||||
|
input_arg {
|
||||||
|
name: "inputs"
|
||||||
|
type_attr: "T"
|
||||||
|
number_attr: "N"
|
||||||
|
}
|
||||||
|
output_arg {
|
||||||
|
name: "output"
|
||||||
|
type_attr: "T"
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "N"
|
||||||
|
type: "int"
|
||||||
|
has_minimum: true
|
||||||
|
minimum: 1
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "T"
|
||||||
|
type: "type"
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "is_mirrored_variable"
|
||||||
|
type: "bool"
|
||||||
|
default_value {
|
||||||
|
b: false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "index"
|
||||||
|
type: "int"
|
||||||
|
default_value {
|
||||||
|
i: -1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "is_packed"
|
||||||
|
type: "bool"
|
||||||
|
default_value {
|
||||||
|
b: false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
@ -49929,6 +49929,13 @@ op {
|
|||||||
i: -1
|
i: -1
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
attr {
|
||||||
|
name: "is_packed"
|
||||||
|
type: "bool"
|
||||||
|
default_value {
|
||||||
|
b: false
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
op {
|
op {
|
||||||
name: "TPUReplicatedOutput"
|
name: "TPUReplicatedOutput"
|
||||||
|
Loading…
Reference in New Issue
Block a user