@tensorflow/tfjs-layers/dist/index.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" />
import './flags_layers';
import '@tensorflow/tfjs-core';
import '@tensorflow/tfjs-core/dist/register_all_gradients';
import * as constraints from './exports_constraints';
import * as initializers from './exports_initializers';
import * as layers from './exports_layers';
import * as metrics from './exports_metrics';
import * as models from './exports_models';
import * as regularizers from './exports_regularizers';
export { CallbackList, CustomCallback, CustomCallbackArgs, History } from './base_callbacks';
export { Callback, callbacks, EarlyStopping, EarlyStoppingCallbackArgs } from './callbacks';
export { InputSpec, SymbolicTensor } from './engine/topology';
export { LayersModel, ModelCompileArgs, ModelEvaluateArgs } from './engine/training';
export { ModelFitDatasetArgs } from './engine/training_dataset';
export { ModelFitArgs } from './engine/training_tensors';
export { ClassWeight, ClassWeightMap } from './engine/training_utils';
export { input, loadLayersModel, model, registerCallbackConstructor, sequential } from './exports';
export { Shape } from './keras_format/common';
export { GRUCellLayerArgs, GRULayerArgs, LSTMCellLayerArgs, LSTMLayerArgs, RNN, RNNLayerArgs, SimpleRNNCellLayerArgs, SimpleRNNLayerArgs } from './layers/recurrent';
export { Logs } from './logs';
export { ModelAndWeightsConfig, Sequential, SequentialArgs } from './models';
export { LayerVariable } from './variables';
export { version as version_layers } from './version';
export { constraints, initializers, layers, metrics, models, regularizers };