index.ts

/**
 * @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.
 * =============================================================================
 */

import './flags_layers';
import '@tensorflow/tfjs-core';
// tslint:disable-next-line: no-imports-from-dist
import '@tensorflow/tfjs-core/dist/register_all_gradients';

// This file lists all exports of TensorFlow.js Layers

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};