@tensorflow/tfjs-core
Version:
Hardware-accelerated JavaScript library for machine intelligence
37 lines (36 loc) • 1.99 kB
TypeScript
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';