1 | "use strict";
|
2 | Object.defineProperty(exports, "__esModule", { value: true });
|
3 |
|
4 |
|
5 |
|
6 |
|
7 |
|
8 |
|
9 |
|
10 |
|
11 |
|
12 |
|
13 |
|
14 |
|
15 |
|
16 |
|
17 |
|
18 |
|
19 |
|
20 |
|
21 |
|
22 | var tfjs_core_1 = require("@tensorflow/tfjs-core");
|
23 | var Dilation2D_1 = require("./kernels/Dilation2D");
|
24 | var Dilation2DBackpropFilter_1 = require("./kernels/Dilation2DBackpropFilter");
|
25 | var Dilation2DBackpropInput_1 = require("./kernels/Dilation2DBackpropInput");
|
26 | var NonMaxSuppressionV4_1 = require("./kernels/NonMaxSuppressionV4");
|
27 | var NonMaxSuppressionV5_1 = require("./kernels/NonMaxSuppressionV5");
|
28 | var Softmax_1 = require("./kernels/Softmax");
|
29 | var SquaredDifference_1 = require("./kernels/SquaredDifference");
|
30 |
|
31 | var kernelConfigs = [
|
32 | Dilation2D_1.dilation2dConfig, Dilation2DBackpropInput_1.dilation2dBackpropInputConfig,
|
33 | Dilation2DBackpropFilter_1.dilation2dBackpropFilterConfig, NonMaxSuppressionV4_1.nonMaxSuppressionV4Config,
|
34 | NonMaxSuppressionV5_1.nonMaxSuppressionV5Config, Softmax_1.softmaxConfig, SquaredDifference_1.squaredDifferenceConfig
|
35 | ];
|
36 | for (var _i = 0, kernelConfigs_1 = kernelConfigs; _i < kernelConfigs_1.length; _i++) {
|
37 | var kernelConfig = kernelConfigs_1[_i];
|
38 | tfjs_core_1.registerKernel(kernelConfig);
|
39 | }
|