@tensorflow/tfjs-layers
Version:
TensorFlow layers API in JavaScript
26 lines (25 loc) • 1.16 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/padding_serialization" />
import { DataFormatSerialization } from '../common';
import { BaseLayerSerialization, LayerConfig } from '../topology_config';
export interface ZeroPadding2DLayerConfig extends LayerConfig {
padding?: number | [number, number] | [[number, number], [number, number]];
data_format?: DataFormatSerialization;
}
export type ZeroPadding2DLayerSerialization = BaseLayerSerialization<'ZeroPadding2D', ZeroPadding2DLayerConfig>;
export type PaddingLayerSerialization = ZeroPadding2DLayerSerialization;
export type PaddingLayerClassName = PaddingLayerSerialization['class_name'];
/**
* A string array of valid PaddingLayer class names.
*
* This is guaranteed to match the `PaddingLayerClassName` union type.
*/
export declare const paddingLayerClassNames: PaddingLayerClassName[];