@tensorflow/tfjs-layers
Version:
TensorFlow layers API in JavaScript
20 lines (19 loc) • 706 B
TypeScript
/**
* @license
* Copyright 2018 Google LLC
*
* Use of this source code is governed by an MIT-style
* license that can be found in the LICENSE file or at
* https://opensource.org/licenses/MIT.
* =============================================================================
*/
/// <amd-module name="@tensorflow/tfjs-layers/dist/utils/variable_utils" />
import { LayerVariable } from '../variables';
/**
* Count the elements in an Array of LayerVariables.
*
* @param weights: The LayerVariables of which the constituent numbers are to
* be counted.
* @returns A count of the elements in all the LayerVariables
*/
export declare function countParamsInWeights(weights: LayerVariable[]): number;