@tensorflow/tfjs-converter/dist/executor/tensor_utils.d.ts

Tensorflow model converter for javascript

/**
 * @license
 * Copyright 2020 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.
 * =============================================================================
 */
/**
 * This differs from util.assertShapesMatch in that it allows values of
 * negative one, an undefined size of a dimensinon, in a shape to match
 * anything.
 */
/// <amd-module name="@tensorflow/tfjs-converter/dist/executor/tensor_utils" />
import { Tensor } from '@tensorflow/tfjs-core';
/**
 * Used by TensorList and TensorArray to verify if elementShape matches, support
 * negative value as the dim shape.
 * @param shapeA
 * @param shapeB
 * @param errorMessagePrefix
 */
export declare function assertShapesMatchAllowUndefinedSize(shapeA: number | number[], shapeB: number | number[], errorMessagePrefix?: string): void;
export declare function fullDefinedShape(elementShape: number | number[]): boolean;
/**
 * Generate the output element shape from the list elementShape, list tensors
 * and input param.
 * @param listElementShape
 * @param tensors
 * @param elementShape
 */
export declare function inferElementShape(listElementShape: number | number[], tensors: Tensor[], elementShape: number | number[]): number[];
export declare function mergeElementShape(elementShapeA: number | number[], elementShapeB: number | number[]): number | number[];