UNPKG

4.93 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': 'FusedBatchNorm',
20 'category': 'normalization',
21 'inputs': [
22 { 'start': 0, 'name': 'x', 'type': 'tensor' },
23 { 'start': 1, 'name': 'scale', 'type': 'tensor' },
24 { 'start': 2, 'name': 'offset', 'type': 'tensor' },
25 { 'start': 3, 'name': 'mean', 'type': 'tensor' },
26 { 'start': 4, 'name': 'variance', 'type': 'tensor' },
27 ],
28 'attrs': [
29 {
30 'tfName': 'epsilon',
31 'name': 'epsilon',
32 'type': 'number',
33 'defaultValue': 0.001
34 },
35 {
36 'tfName': 'data_format',
37 'name': 'dataFormat',
38 'type': 'string',
39 'notSupported': true
40 }
41 ]
42 },
43 {
44 'tfOpName': 'FusedBatchNormV2',
45 'category': 'normalization',
46 'inputs': [
47 { 'start': 0, 'name': 'x', 'type': 'tensor' },
48 { 'start': 1, 'name': 'scale', 'type': 'tensor' },
49 { 'start': 2, 'name': 'offset', 'type': 'tensor' },
50 { 'start': 3, 'name': 'mean', 'type': 'tensor' },
51 { 'start': 4, 'name': 'variance', 'type': 'tensor' },
52 ],
53 'attrs': [
54 {
55 'tfName': 'epsilon',
56 'name': 'epsilon',
57 'type': 'number',
58 'defaultValue': 0.001
59 },
60 {
61 'tfName': 'data_format',
62 'name': 'dataFormat',
63 'type': 'string',
64 'notSupported': true
65 }
66 ]
67 },
68 {
69 'tfOpName': 'FusedBatchNormV3',
70 'category': 'normalization',
71 'inputs': [
72 { 'start': 0, 'name': 'x', 'type': 'tensor' },
73 { 'start': 1, 'name': 'scale', 'type': 'tensor' },
74 { 'start': 2, 'name': 'offset', 'type': 'tensor' },
75 { 'start': 3, 'name': 'mean', 'type': 'tensor' },
76 { 'start': 4, 'name': 'variance', 'type': 'tensor' },
77 ],
78 'attrs': [
79 {
80 'tfName': 'epsilon',
81 'name': 'epsilon',
82 'type': 'number',
83 'defaultValue': 0.001
84 },
85 {
86 'tfName': 'data_format',
87 'name': 'dataFormat',
88 'type': 'string',
89 'notSupported': true
90 }
91 ]
92 },
93 {
94 'tfOpName': 'LRN',
95 'category': 'normalization',
96 'inputs': [
97 { 'start': 0, 'name': 'x', 'type': 'tensor' },
98 ],
99 'attrs': [
100 {
101 'tfName': 'depth_radius',
102 'name': 'radius',
103 'type': 'number',
104 'defaultValue': 5
105 },
106 { 'tfName': 'bias', 'name': 'bias', 'type': 'number', 'defaultValue': 1.0 },
107 {
108 'tfName': 'alpha',
109 'name': 'alpha',
110 'type': 'number',
111 'defaultValue': 1.0
112 },
113 {
114 'tfName': 'beta',
115 'name': 'beta',
116 'type': 'number',
117 'defaultValue': 0.5
118 }
119 ]
120 },
121 {
122 'tfOpName': 'Softmax',
123 'category': 'normalization',
124 'inputs': [{ 'start': 0, 'name': 'x', 'type': 'tensor' }]
125 },
126 {
127 'tfOpName': 'LogSoftmax',
128 'category': 'normalization',
129 'inputs': [{ 'start': 0, 'name': 'x', 'type': 'tensor' }]
130 },
131 {
132 'tfOpName': 'SparseToDense',
133 'category': 'normalization',
134 'inputs': [
135 { 'start': 0, 'name': 'sparseIndices', 'type': 'tensor' },
136 { 'start': 1, 'name': 'outputShape', 'type': 'number[]' },
137 { 'start': 2, 'name': 'sparseValues', 'type': 'tensor' },
138 { 'start': 3, 'name': 'defaultValue', 'type': 'tensor' },
139 ],
140 'attrs': [{
141 'tfName': 'validate_indices',
142 'name': 'validateIndices',
143 'type': 'bool',
144 'defaultValue': true,
145 'notSupported': true
146 }]
147 }
148];
149//# sourceMappingURL=normalization.js.map
\No newline at end of file