@tensorflow/tfjs-layers
Version:
TensorFlow layers API in JavaScript
34 lines (33 loc) • 1.78 kB
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/keras_format/layers/merge_serialization" />
import { BaseLayerSerialization, LayerConfig } from '../topology_config';
export type AddLayerSerialization = BaseLayerSerialization<'Add', LayerConfig>;
export type MultiplyLayerSerialization = BaseLayerSerialization<'Multiply', LayerConfig>;
export type AverageLayerSerialization = BaseLayerSerialization<'Average', LayerConfig>;
export type MaximumLayerSerialization = BaseLayerSerialization<'Maximum', LayerConfig>;
export type MinimumLayerSerialization = BaseLayerSerialization<'Minimum', LayerConfig>;
export interface ConcatenateLayerConfig extends LayerConfig {
axis?: number;
}
export type ConcatenateLayerSerialization = BaseLayerSerialization<'Concatenate', ConcatenateLayerConfig>;
export interface DotLayerConfig extends LayerConfig {
axes: number | [number, number];
normalize?: boolean;
}
export type DotLayerSerialization = BaseLayerSerialization<'Dot', DotLayerConfig>;
export type MergeLayerSerialization = AddLayerSerialization | MultiplyLayerSerialization | AverageLayerSerialization | MaximumLayerSerialization | MinimumLayerSerialization | ConcatenateLayerSerialization | DotLayerSerialization;
export type MergeLayerClassName = MergeLayerSerialization['class_name'];
/**
* A string array of valid MergeLayer class names.
*
* This is guaranteed to match the `MergeLayerClassName` union type.
*/
export declare const mergeLayerClassNames: MergeLayerClassName[];