UNPKG

@tensorflow/tfjs-core

Version:

Hardware-accelerated JavaScript library for machine intelligence

37 lines (36 loc) 1.99 kB
import './kernels/backend_webgl'; import './kernels/backend_cpu'; import { BrowserUtil } from './browser_util'; import * as environment from './environment'; import * as io from './io/io'; import * as serialization from './serialization'; import * as test_util from './test_util'; import * as util from './util'; import { version } from './version'; import * as webgl from './webgl'; export { AdadeltaOptimizer } from './optimizers/adadelta_optimizer'; export { AdagradOptimizer } from './optimizers/adagrad_optimizer'; export { AdamOptimizer } from './optimizers/adam_optimizer'; export { AdamaxOptimizer } from './optimizers/adamax_optimizer'; export { MomentumOptimizer } from './optimizers/momentum_optimizer'; export { Optimizer } from './optimizers/optimizer'; export { RMSPropOptimizer } from './optimizers/rmsprop_optimizer'; export { SGDOptimizer } from './optimizers/sgd_optimizer'; export { Scalar, Tensor, Tensor1D, Tensor2D, Tensor3D, Tensor4D, TensorBuffer, variable, Variable } from './tensor'; export { DataType, InferenceModel, ModelPredictConfig, NamedTensorMap, Rank, ShapeMap } from './types'; export * from './ops/ops'; export { LSTMCellFunc } from './ops/lstm'; export { Reduction } from './ops/loss_ops'; export * from './train'; export * from './globals'; export { ENV, Environment, Features } from './environment'; export declare const setBackend: typeof environment.Environment.setBackend; export declare const getBackend: typeof environment.Environment.getBackend; export declare const disposeVariables: typeof environment.Environment.disposeVariables; export declare const memory: typeof environment.Environment.memory; export { TimingInfo } from './engine'; export { version as version_core }; export { doc } from './doc'; export declare const nextFrame: typeof BrowserUtil.nextFrame; export { environment, io, serialization, test_util, util, webgl }; export { KernelBackend, BackendTimingInfo } from './kernels/backend';