@tensorflow/tfjs-layers
Version:
TensorFlow layers API in JavaScript
32 lines (31 loc) • 1.51 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/wrappers_serialization" />
import { BidirectionalMergeMode } from '../common';
import { BaseLayerSerialization, LayerConfig } from '../topology_config';
import { LayerSerialization } from './layer_serialization';
import { RecurrentLayerSerialization } from './recurrent_serialization';
export type TimeDistributedLayerSerialization = BaseLayerSerialization<'TimeDistributed', TimeDistributedLayerConfig>;
export interface TimeDistributedLayerConfig extends LayerConfig {
layer: LayerSerialization;
}
export type BidirectionalLayerSerialization = BaseLayerSerialization<'Bidirectional', BidirectionalLayerConfig>;
export interface BidirectionalLayerConfig extends LayerConfig {
layer: RecurrentLayerSerialization;
merge_mode?: BidirectionalMergeMode;
}
export type WrapperLayerSerialization = TimeDistributedLayerSerialization | BidirectionalLayerSerialization;
export type WrapperLayerClassName = WrapperLayerSerialization['class_name'];
/**
* A string array of valid WrapperLayer class names.
*
* This is guaranteed to match the `WrapperLayerClassName` union type.
*/
export declare const wrapperLayerClassNames: WrapperLayerClassName[];