/**

* @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.

* =============================================================================

*/

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 function countParamsInWeights(weights: LayerVariable[]): number {

let count = 0;

for (const weight of weights) {

if (weight.shape.length === 0) {

count += 1;

} else {

count += weight.shape.reduce((a, b) => a * b);

}

}

return count;

}