UNPKG

@tensorflow/tfjs-core

Version:

Hardware-accelerated JavaScript library for machine intelligence

29 lines 1.59 kB
"use strict"; var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) { var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d; if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc); else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r; return c > 3 && r && Object.defineProperty(target, key, r), r; }; Object.defineProperty(exports, "__esModule", { value: true }); var doc_1 = require("../doc"); var environment_1 = require("../environment"); var util = require("../util"); var operation_1 = require("./operation"); var StridedSliceOps = (function () { function StridedSliceOps() { } StridedSliceOps.stridedSlice = function (x, begin, end, strides, beginMask, endMask) { if (beginMask === void 0) { beginMask = 0; } if (endMask === void 0) { endMask = 0; } util.assertArgumentsAreTensors({ x: x }, 'stridedSlice'); return environment_1.ENV.engine.runKernel(function (backend) { return backend.stridedSlice(x, begin, end, strides, beginMask, endMask); }, { x: x }); }; __decorate([ doc_1.doc({ heading: 'Operations', subheading: 'Slicing and Joining' }), operation_1.operation ], StridedSliceOps, "stridedSlice", null); return StridedSliceOps; }()); exports.StridedSliceOps = StridedSliceOps; //# sourceMappingURL=strided_slice.js.map