/** * @license * Copyright 2018 Google LLC. All Rights Reserved. * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. * ============================================================================= */ /// /** * Gets the new shape of the input Tensor after it's been reshaped * to: * [blockShape[0], ..., blockShape[M-1], batch / prod(blockShape), * inputShape[1], ..., inputShape[N-1]] * * See step 1: https://www.tensorflow.org/api_docs/python/tf/batch_to_space_nd */ export declare function getReshaped(inputShape: number[], blockShape: number[], prod: number, batchToSpace?: boolean): number[]; /** * Gets the permutation that will transpose the dimensions of the * reshaped tensor to shape: * * [batch / prod(block_shape),inputShape[1], blockShape[0], ..., * inputShape[M], blockShape[M-1],inputShape[M+1], ..., inputShape[N-1]] * * see step 2: https://www.tensorflow.org/api_docs/python/tf/batch_to_space_nd */ export declare function getPermuted(reshapedRank: number, blockShapeRank: number, batchToSpace?: boolean): number[]; /** * Gets the shape of the reshaped and permuted input Tensor before any cropping * is applied. The new shape will be: * * [batch / prod(blockShape),inputShape[1] * blockShape[0], ..., * inputShape[M] * blockShape[M-1],inputShape[M+1], ..., inputShape[N-1]] * * See step 3: https://www.tensorflow.org/api_docs/python/tf/batch_to_space_nd */ export declare function getReshapedPermuted(inputShape: number[], blockShape: number[], prod: number, batchToSpace?: boolean): number[]; /** * Converts the crops argument into the beginning coordinates of a slice * operation. */ export declare function getSliceBeginCoords(crops: number[][], blockShape: number): number[]; /** * Converts the crops argument into the size of a slice operation. When * combined with getSliceBeginCoords this function allows the reshaped and * permuted Tensor to be cropped to its final output shape of: * * inputShape[1] * blockShape[0] - crops[0,0] - crops[0,1], ..., * inputShape[M] * blockShape[M-1] -crops[M-1,0] - * crops[M-1,1],inputShape[M+1], ..., inputShape[N-1]] * * See step 4: https://www.tensorflow.org/api_docs/python/tf/batch_to_space_nd */ export declare function getSliceSize(uncroppedShape: number[], crops: number[][], blockShape: number): number[];