diff --git a/src/backends/backend_util.ts b/src/backends/backend_util.ts index 34ab4f75e7..3de503109d 100644 --- a/src/backends/backend_util.ts +++ b/src/backends/backend_util.ts @@ -28,7 +28,9 @@ export * from '../ops/axis_util'; export * from '../ops/broadcast_util'; export * from '../ops/concat_util'; export * from '../ops/conv_util'; -export {TypedArray, upcastType, PixelData} from '../types'; +export {Activation} from '../ops/fused_util'; + +export {BackendValues, TypedArray, upcastType, PixelData} from '../types'; export {MemoryInfo, TimingInfo} from '../engine'; export function castTensor( diff --git a/src/index.ts b/src/index.ts index 15c34a75af..453925df6f 100644 --- a/src/index.ts +++ b/src/index.ts @@ -53,7 +53,7 @@ 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';