@tensorflow/tfjs-layers/dist/keras_format/input_config.d.ts

TensorFlow layers API in JavaScript

/**
 * @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/input_config" />
import { DataType } from '@tensorflow/tfjs-core';
import { Shape } from './common';
import { BaseLayerSerialization } from './topology_config';
export type InputLayerConfig = {
    name?: string;
    input_shape?: Shape;
    batch_size?: number;
    batch_input_shape?: Shape;
    dtype?: DataType;
    sparse?: boolean;
};
export type InputLayerSerialization = BaseLayerSerialization<'InputLayer', InputLayerConfig>;
export type InputLayerClassName = InputLayerSerialization['class_name'];
/**
 * A string array of valid InputLayer class names.
 *
 * This is guaranteed to match the `InputLayerClassName` union type.
 */
export declare const inputLayerClassNames: InputLayerClassName[];