UNPKG

7.3 kBJavaScriptView Raw
1/**
2 * @license
3 * Copyright 2018 Google LLC. All Rights Reserved.
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 * =============================================================================
16 */
17export const json = [
18 {
19 'tfOpName': 'ConcatV2',
20 'category': 'slice_join',
21 'inputs': [
22 { 'start': 0, 'end': -1, 'name': 'tensors', 'type': 'tensors' },
23 { 'start': -1, 'name': 'axis', 'type': 'number' }
24 ],
25 'attrs': [{ 'tfName': 'N', 'name': 'n', 'type': 'number', 'defaultValue': 2 }]
26 },
27 {
28 'tfOpName': 'Concat',
29 'category': 'slice_join',
30 'inputs': [
31 { 'start': 1, 'end': 0, 'name': 'tensors', 'type': 'tensors' },
32 { 'start': 0, 'name': 'axis', 'type': 'number' }
33 ],
34 'attrs': [{ 'tfName': 'N', 'name': 'n', 'type': 'number', 'defaultValue': 2 }]
35 },
36 {
37 'tfOpName': 'GatherV2',
38 'category': 'slice_join',
39 'inputs': [
40 { 'start': 0, 'name': 'x', 'type': 'tensor' },
41 { 'start': 1, 'name': 'indices', 'type': 'tensor' },
42 { 'start': 2, 'name': 'axis', 'type': 'number', 'defaultValue': 0 }
43 ],
44 'attrs': [{
45 'tfName': 'batch_dims',
46 'name': 'batchDims',
47 'type': 'number',
48 'defaultValue': 0
49 }]
50 },
51 {
52 'tfOpName': 'Gather',
53 'category': 'slice_join',
54 'inputs': [
55 { 'start': 0, 'name': 'x', 'type': 'tensor' },
56 { 'start': 1, 'name': 'indices', 'type': 'tensor' },
57 ],
58 'attrs': [{
59 'tfName': 'validate_indices',
60 'name': 'validateIndices',
61 'type': 'bool',
62 'notSupported': true
63 }]
64 },
65 {
66 'tfOpName': 'Reverse',
67 'category': 'slice_join',
68 'inputs': [
69 { 'start': 0, 'name': 'x', 'type': 'tensor' },
70 { 'start': 1, 'name': 'dims', 'type': 'bool[]' }
71 ]
72 },
73 {
74 'tfOpName': 'ReverseV2',
75 'category': 'slice_join',
76 'inputs': [
77 { 'start': 0, 'name': 'x', 'type': 'tensor' },
78 { 'start': 1, 'name': 'axis', 'type': 'number[]' }
79 ]
80 },
81 {
82 'tfOpName': 'Slice',
83 'category': 'slice_join',
84 'inputs': [
85 { 'start': 0, 'name': 'x', 'type': 'tensor' },
86 { 'start': 1, 'name': 'begin', 'type': 'number[]' },
87 { 'start': 2, 'name': 'size', 'type': 'number[]' }
88 ]
89 },
90 {
91 'tfOpName': 'StridedSlice',
92 'category': 'slice_join',
93 'inputs': [
94 { 'start': 0, 'name': 'x', 'type': 'tensor' },
95 { 'start': 1, 'name': 'begin', 'type': 'number[]' },
96 { 'start': 2, 'name': 'end', 'type': 'number[]' },
97 { 'start': 3, 'name': 'strides', 'type': 'number[]' },
98 ],
99 'attrs': [
100 {
101 'tfName': 'begin_mask',
102 'name': 'beginMask',
103 'type': 'number',
104 'defaultValue': 0
105 },
106 {
107 'tfName': 'end_mask',
108 'name': 'endMask',
109 'type': 'number',
110 'defaultValue': 0
111 },
112 {
113 'tfName': 'new_axis_mask',
114 'name': 'newAxisMask',
115 'type': 'number',
116 'defaultValue': 0
117 },
118 {
119 'tfName': 'ellipsis_mask',
120 'name': 'ellipsisMask',
121 'type': 'number',
122 'defaultValue': 0
123 },
124 {
125 'tfName': 'shrink_axis_mask',
126 'name': 'shrinkAxisMask',
127 'type': 'number',
128 'defaultValue': 0
129 }
130 ]
131 },
132 {
133 'tfOpName': 'Pack',
134 'category': 'slice_join',
135 'inputs': [
136 { 'start': 0, 'end': 0, 'name': 'tensors', 'type': 'tensors' },
137 ],
138 'attrs': [
139 { 'tfName': 'axis', 'name': 'axis', 'type': 'number', 'defaultValue': 0 }
140 ]
141 },
142 {
143 'tfOpName': 'Unpack',
144 'category': 'slice_join',
145 'inputs': [
146 { 'start': 0, 'name': 'tensor', 'type': 'tensor' },
147 ],
148 'attrs': [
149 { 'tfName': 'axis', 'name': 'axis', 'type': 'number', 'defaultValue': 0 }, {
150 'tfName': 'num',
151 'name': 'num',
152 'type': 'number',
153 'defaultValue': 0,
154 'notSupported': true
155 }
156 ]
157 },
158 {
159 'tfOpName': 'Tile',
160 'category': 'slice_join',
161 'inputs': [
162 { 'start': 0, 'name': 'x', 'type': 'tensor' },
163 { 'start': 1, 'name': 'reps', 'type': 'number[]' }
164 ]
165 },
166 {
167 'tfOpName': 'Split',
168 'category': 'slice_join',
169 'inputs': [
170 { 'start': 0, 'name': 'axis', 'type': 'number', 'defaultValue': 0 },
171 { 'start': 1, 'name': 'x', 'type': 'tensor' },
172 ],
173 'attrs': [{
174 'tfName': 'num_split',
175 'name': 'numOrSizeSplits',
176 'type': 'number',
177 'defaultValue': 1
178 }]
179 },
180 {
181 'tfOpName': 'SplitV',
182 'category': 'slice_join',
183 'inputs': [
184 { 'start': 0, 'name': 'x', 'type': 'tensor' },
185 { 'start': 1, 'name': 'numOrSizeSplits', 'type': 'number[]' },
186 { 'start': 2, 'name': 'axis', 'type': 'number', 'defaultValue': 0 }
187 ]
188 },
189 {
190 'tfOpName': 'ScatterNd',
191 'category': 'slice_join',
192 'inputs': [
193 { 'start': 0, 'name': 'indices', 'type': 'tensor' },
194 { 'start': 1, 'name': 'values', 'type': 'tensor' },
195 { 'start': 2, 'name': 'shape', 'type': 'number[]' }
196 ]
197 },
198 {
199 'tfOpName': 'GatherNd',
200 'category': 'slice_join',
201 'inputs': [
202 { 'start': 0, 'name': 'x', 'type': 'tensor' },
203 { 'start': 1, 'name': 'indices', 'type': 'tensor' }
204 ]
205 },
206 {
207 'tfOpName': 'SparseToDense',
208 'category': 'slice_join',
209 'inputs': [
210 { 'start': 0, 'name': 'sparseIndices', 'type': 'tensor' },
211 { 'start': 1, 'name': 'outputShape', 'type': 'number[]' },
212 { 'start': 2, 'name': 'sparseValues', 'type': 'tensor' },
213 { 'start': 3, 'name': 'defaultValue', 'type': 'tensor' },
214 ],
215 'attrs': [{
216 'tfName': 'validate_indices',
217 'name': 'validateIndices',
218 'type': 'bool',
219 'defaultValue': false,
220 'notSupported': true
221 }]
222 }
223];
224//# sourceMappingURL=slice_join.js.map
\No newline at end of file