1 | "use strict";
|
2 |
|
3 |
|
4 |
|
5 |
|
6 |
|
7 |
|
8 |
|
9 |
|
10 |
|
11 |
|
12 |
|
13 |
|
14 |
|
15 |
|
16 |
|
17 |
|
18 | Object.defineProperty(exports, "__esModule", { value: true });
|
19 | var tfjs_1 = require("@tensorflow/tfjs");
|
20 | var nodejs_kernel_backend_1 = require("../nodejs_kernel_backend");
|
21 | exports.dilation2dBackpropFilterConfig = {
|
22 | kernelName: tfjs_1.Dilation2DBackpropFilter,
|
23 | backendName: 'tensorflow',
|
24 | kernelFunc: function (_a) {
|
25 | var inputs = _a.inputs, backend = _a.backend, attrs = _a.attrs;
|
26 | var _b = inputs, x = _b.x, filter = _b.filter, dy = _b.dy;
|
27 | var _c = attrs, strides = _c.strides, pad = _c.pad, dilations = _c.dilations;
|
28 | var _d = tfjs_1.backend_util.computeDilation2DInfo(x.shape, filter.shape, strides, pad, 'NHWC' , dilations), dilationHeight = _d.dilationHeight, dilationWidth = _d.dilationWidth, padInfo = _d.padInfo, strideHeight = _d.strideHeight, strideWidth = _d.strideWidth;
|
29 | var $strides = [1, strideHeight, strideWidth, 1];
|
30 | var $dilations = [1, dilationHeight, dilationWidth, 1];
|
31 | var nodeBackend = backend;
|
32 | var opAttrs = [
|
33 | nodejs_kernel_backend_1.createTensorsTypeOpAttr('T', x.dtype),
|
34 | { name: 'strides', type: nodeBackend.binding.TF_ATTR_INT, value: $strides },
|
35 | { name: 'rates', type: nodeBackend.binding.TF_ATTR_INT, value: $dilations },
|
36 | {
|
37 | name: 'padding',
|
38 | type: nodeBackend.binding.TF_ATTR_STRING,
|
39 | value: padInfo.type
|
40 | }
|
41 | ];
|
42 | return nodeBackend.executeSingleOutput(tfjs_1.Dilation2DBackpropFilter, opAttrs, [x, filter, dy]);
|
43 | }
|
44 | };
|