1 |
|
2 |
|
3 |
|
4 |
|
5 |
|
6 |
|
7 |
|
8 |
|
9 |
|
10 |
|
11 |
|
12 |
|
13 |
|
14 |
|
15 |
|
16 |
|
17 |
|
18 | import { TensorInfo } from '../kernel_registry';
|
19 | export declare type SliceInfo = {
|
20 | nonStrided: boolean;
|
21 | $begin: number[];
|
22 | $end: number[];
|
23 | $strides: number[];
|
24 | size: number[];
|
25 | newShape: number[];
|
26 | outShape: number[];
|
27 | };
|
28 | export declare function assertParamsValid(input: TensorInfo, begin: number[], size: number[]): void;
|
29 |
|
30 | export declare function maskToAxes(mask: number): number[];
|
31 |
|
32 | export declare function computeOutShape(begin: number[], end: number[], strides: number[]): number[];
|
33 | export declare function stridesWithElidedDims(strides: number[], ellipsisInsertionIndex: number, numElidedAxes: number, inputShape: number[]): number[];
|
34 | export declare function getNormalizedAxes(inputShape: number[], ellipsisAxes: number[], numInterpolatedAxes: number, begin: number[], end: number[], strides: number[], beginMask: number, endMask: number, ellipsisMask: number): {
|
35 | begin: number[];
|
36 | end: number[];
|
37 | strides: number[];
|
38 | };
|
39 | export declare function startIndicesWithElidedDims(beginMask: number, ellipsisInsertionIndex: number, numElidedAxes: number, originalBegin: number[], inputShape: number[]): number[];
|
40 | export declare function stopIndicesWithElidedDims(endMask: number, ellipsisInsertionIndex: number, numElidedAxes: number, originalEnd: number[], inputShape: number[]): number[];
|
41 | export declare function stridesForAxis(strides: number[], axis: number, ellipsisMask: number): number;
|
42 | export declare function startForAxis(beginMask: number, startIndices: number[], strides: number[], inputShape: number[], axis: number, ellipsisMask: number): number;
|
43 | export declare function stopForAxis(endMask: number, stopIndices: number[], strides: number[], inputShape: number[], axis: number, ellipsisMask: number): number;
|
44 |
|
45 |
|
46 |
|
47 |
|
48 | export declare function isSliceContinous(shape: number[], begin: number[], size: number[]): boolean;
|
49 | export declare function computeFlatOffset(begin: number[], strides: number[]): number;
|
50 | export declare function parseSliceParams(x: TensorInfo, begin: number | number[], size?: number | number[]): number[][];
|
51 | export declare function sliceInfo(xShape: number[], begin: number[], end: number[], strides: number[], beginMask: number, endMask: number, ellipsisMask: number, newAxisMask: number, shrinkAxisMask: number): SliceInfo;
|