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
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "is_packed"
|
||||
type: "bool"
|
||||
default_value {
|
||||
b: false
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "TPUReplicatedOutput"
|
||||
|
Loading…
Reference in New Issue
Block a user