"use strict"; /** * @license * Copyright 2017 Google Inc. All Rights Reserved. * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. * ============================================================================= */ function __export(m) { for (var p in m) if (!exports.hasOwnProperty(p)) exports[p] = m[p]; } Object.defineProperty(exports, "__esModule", { value: true }); /** * @fileoverview * @suppress {partialAlias} Optimization disabled due to passing the module * object into a function below: * * import * as ops from './ops/ops'; * setOpHandler(ops); */ // Engine is the global singleton that needs to be initialized before the rest // of the app. require("./engine"); // Register backend-agnostic flags. require("./flags"); // backend_cpu.ts and backend_webgl.ts are standalone files and should be // explicitly included here. require("./backends/webgl/backend_webgl"); require("./backends/cpu/backend_cpu"); // Import all kernels from cpu. require("./backends/cpu/register_all_kernels"); // Import all kernels from webgl. require("./backends/webgl/register_all_kernels"); // Register all the gradients. require("./register_all_gradients"); require("./platforms/platform_browser"); require("./platforms/platform_node"); var backend_util = require("./backends/backend_util"); exports.backend_util = backend_util; // Serialization. var io = require("./io/io"); exports.io = io; var math = require("./math"); exports.math = math; var browser = require("./ops/browser"); exports.browser = browser; var gather_util = require("./ops/gather_nd_util"); exports.gather_util = gather_util; var scatter_util = require("./ops/scatter_nd_util"); exports.scatter_util = scatter_util; var slice_util = require("./ops/slice_util"); exports.slice_util = slice_util; var serialization = require("./serialization"); exports.serialization = serialization; var tensor_1 = require("./tensor"); var tensor_util = require("./tensor_util"); exports.tensor_util = tensor_util; var test_util = require("./test_util"); exports.test_util = test_util; var util = require("./util"); exports.util = util; var version_1 = require("./version"); exports.version_core = version_1.version; var webgl = require("./webgl"); exports.webgl = webgl; // Optimizers. var adadelta_optimizer_1 = require("./optimizers/adadelta_optimizer"); exports.AdadeltaOptimizer = adadelta_optimizer_1.AdadeltaOptimizer; var adagrad_optimizer_1 = require("./optimizers/adagrad_optimizer"); exports.AdagradOptimizer = adagrad_optimizer_1.AdagradOptimizer; var adam_optimizer_1 = require("./optimizers/adam_optimizer"); exports.AdamOptimizer = adam_optimizer_1.AdamOptimizer; var adamax_optimizer_1 = require("./optimizers/adamax_optimizer"); exports.AdamaxOptimizer = adamax_optimizer_1.AdamaxOptimizer; var momentum_optimizer_1 = require("./optimizers/momentum_optimizer"); exports.MomentumOptimizer = momentum_optimizer_1.MomentumOptimizer; var optimizer_1 = require("./optimizers/optimizer"); exports.Optimizer = optimizer_1.Optimizer; var rmsprop_optimizer_1 = require("./optimizers/rmsprop_optimizer"); exports.RMSPropOptimizer = rmsprop_optimizer_1.RMSPropOptimizer; var sgd_optimizer_1 = require("./optimizers/sgd_optimizer"); exports.SGDOptimizer = sgd_optimizer_1.SGDOptimizer; var tensor_2 = require("./tensor"); exports.Tensor = tensor_2.Tensor; exports.TensorBuffer = tensor_2.TensorBuffer; exports.Variable = tensor_2.Variable; var types_1 = require("./types"); exports.Rank = types_1.Rank; exports.sumOutType = types_1.sumOutType; __export(require("./ops/ops")); var loss_ops_1 = require("./ops/loss_ops"); exports.Reduction = loss_ops_1.Reduction; __export(require("./train")); __export(require("./globals")); __export(require("./kernel_registry")); var gradients_1 = require("./gradients"); exports.customGrad = gradients_1.customGrad; exports.grad = gradients_1.grad; exports.grads = gradients_1.grads; exports.valueAndGrad = gradients_1.valueAndGrad; exports.valueAndGrads = gradients_1.valueAndGrads; exports.variableGrads = gradients_1.variableGrads; var environment_1 = require("./environment"); exports.Environment = environment_1.Environment; exports.env = environment_1.env; exports.ENV = environment_1.ENV; // Top-level method exports. var browser_util_1 = require("./browser_util"); exports.nextFrame = browser_util_1.nextFrame; // Backend specific. var backend_1 = require("./backends/backend"); exports.KernelBackend = backend_1.KernelBackend; exports.DataStorage = backend_1.DataStorage; var ops = require("./ops/ops"); tensor_1.setOpHandler(ops); // Import all op chainers and add type info to Tensor. require("./public/chained_ops/register_all_chained_ops"); //# sourceMappingURL=index.js.map