Skip to content
This repository was archived by the owner on Aug 15, 2019. It is now read-only.
Merged
Changes from 4 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 6 additions & 2 deletions src/index.ts
Original file line number Diff line number Diff line change
Expand Up @@ -53,11 +53,13 @@ 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 {Scalar, Tensor, Tensor1D, Tensor2D, Tensor3D, Tensor4D, Tensor5D, TensorBuffer, variable, Variable} from './tensor';
export {GradSaveFunc, NamedTensorMap, TensorContainer, TensorContainerArray, TensorContainerObject} from './tensor_types';
export {DataType, DataTypeMap, DataValues, Rank, ShapeMap, TensorLike} from './types';
export {BackendValues, DataType, DataTypeMap, DataValues, Rank, ShapeMap, TensorLike, upcastType} from './types';

export * from './ops/ops';
export {Conv2DInfo, Conv3DInfo} from './ops/conv_util';
export {Activation} from './ops/fused_util';
export {LSTMCellFunc} from './ops/lstm';
export {Reduction} from './ops/loss_ops';

Expand All @@ -69,6 +71,8 @@ export {TimingInfo} from './engine';
export {ENV, Environment} from './environment';
export {Platform} from './platforms/platform';

export {EPSILON_FLOAT32} from './backends/backend';

export {version as version_core};

// Top-level method exports.
Expand Down