Skip to content

Commit 05e7d51

Browse files
tensorflower-gardenertensorflower-gardener
committed
Update ops-related pbtxt files.
PiperOrigin-RevId: 411896036 Change-Id: Ie482fb10c4b23e10fad85fb085a46a621ec865ce
1 parent 661f581 commit 05e7d51

File tree

2 files changed

+160
-0
lines changed

2 files changed

+160
-0
lines changed

‎tensorflow/core/ops/compat/ops_history_v2/DynamicEnqueueTPUEmbeddingArbitraryTensorBatch.pbtxt‎

Lines changed: 80 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,80 @@
1+
op {
2+
name: "DynamicEnqueueTPUEmbeddingArbitraryTensorBatch"
3+
input_arg {
4+
name: "sample_indices_or_row_lengths"
5+
type_attr: "T1"
6+
number_attr: "N"
7+
}
8+
input_arg {
9+
name: "embedding_indices"
10+
type_attr: "T2"
11+
number_attr: "N"
12+
}
13+
input_arg {
14+
name: "aggregation_weights"
15+
type_attr: "T3"
16+
number_attr: "N"
17+
}
18+
input_arg {
19+
name: "mode_override"
20+
type: DT_STRING
21+
}
22+
input_arg {
23+
name: "device_ordinal"
24+
type: DT_INT32
25+
}
26+
attr {
27+
name: "T1"
28+
type: "type"
29+
default_value {
30+
type: DT_INT32
31+
}
32+
allowed_values {
33+
list {
34+
type: DT_INT32
35+
type: DT_INT64
36+
}
37+
}
38+
}
39+
attr {
40+
name: "T2"
41+
type: "type"
42+
default_value {
43+
type: DT_INT32
44+
}
45+
allowed_values {
46+
list {
47+
type: DT_INT32
48+
type: DT_INT64
49+
}
50+
}
51+
}
52+
attr {
53+
name: "T3"
54+
type: "type"
55+
default_value {
56+
type: DT_FLOAT
57+
}
58+
allowed_values {
59+
list {
60+
type: DT_FLOAT
61+
type: DT_DOUBLE
62+
}
63+
}
64+
}
65+
attr {
66+
name: "N"
67+
type: "int"
68+
has_minimum: true
69+
minimum: 1
70+
}
71+
attr {
72+
name: "combiners"
73+
type: "list(string)"
74+
default_value {
75+
list {
76+
}
77+
}
78+
}
79+
is_stateful: true
80+
}

‎tensorflow/core/ops/ops.pbtxt‎

Lines changed: 80 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14187,6 +14187,86 @@ op {
1418714187
}
1418814188
is_stateful: true
1418914189
}
14190+
op {
14191+
name: "DynamicEnqueueTPUEmbeddingArbitraryTensorBatch"
14192+
input_arg {
14193+
name: "sample_indices_or_row_lengths"
14194+
type_attr: "T1"
14195+
number_attr: "N"
14196+
}
14197+
input_arg {
14198+
name: "embedding_indices"
14199+
type_attr: "T2"
14200+
number_attr: "N"
14201+
}
14202+
input_arg {
14203+
name: "aggregation_weights"
14204+
type_attr: "T3"
14205+
number_attr: "N"
14206+
}
14207+
input_arg {
14208+
name: "mode_override"
14209+
type: DT_STRING
14210+
}
14211+
input_arg {
14212+
name: "device_ordinal"
14213+
type: DT_INT32
14214+
}
14215+
attr {
14216+
name: "T1"
14217+
type: "type"
14218+
default_value {
14219+
type: DT_INT32
14220+
}
14221+
allowed_values {
14222+
list {
14223+
type: DT_INT32
14224+
type: DT_INT64
14225+
}
14226+
}
14227+
}
14228+
attr {
14229+
name: "T2"
14230+
type: "type"
14231+
default_value {
14232+
type: DT_INT32
14233+
}
14234+
allowed_values {
14235+
list {
14236+
type: DT_INT32
14237+
type: DT_INT64
14238+
}
14239+
}
14240+
}
14241+
attr {
14242+
name: "T3"
14243+
type: "type"
14244+
default_value {
14245+
type: DT_FLOAT
14246+
}
14247+
allowed_values {
14248+
list {
14249+
type: DT_FLOAT
14250+
type: DT_DOUBLE
14251+
}
14252+
}
14253+
}
14254+
attr {
14255+
name: "N"
14256+
type: "int"
14257+
has_minimum: true
14258+
minimum: 1
14259+
}
14260+
attr {
14261+
name: "combiners"
14262+
type: "list(string)"
14263+
default_value {
14264+
list {
14265+
}
14266+
}
14267+
}
14268+
is_stateful: true
14269+
}
1419014270
op {
1419114271
name: "DynamicPartition"
1419214272
input_arg {

0 commit comments

Comments
 (0)