/**
|
* @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[];
|