Skip to content
This repository was archived by the owner on Aug 15, 2019. It is now read-only.

Commit eb2e989

Browse files
WenheLINikhil Thorat
authored and
Nikhil Thorat
committed
Add exports to remove importing from dist in other packages. (#1856)
DEV This pr is to fix dist importing happens in tfjs-node. A corresponding pr is here, tensorflow/tfjs-node#289
1 parent 893ba6e commit eb2e989

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

src/backends/backend_util.ts

+3-1
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,9 @@ export * from '../ops/axis_util';
2828
export * from '../ops/broadcast_util';
2929
export * from '../ops/concat_util';
3030
export * from '../ops/conv_util';
31-
export {TypedArray, upcastType, PixelData} from '../types';
31+
export {Activation} from '../ops/fused_util';
32+
33+
export {BackendValues, TypedArray, upcastType, PixelData} from '../types';
3234
export {MemoryInfo, TimingInfo} from '../engine';
3335

3436
export function castTensor<T extends Tensor>(

src/index.ts

+1-1
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ export {MomentumOptimizer} from './optimizers/momentum_optimizer';
5353
export {Optimizer} from './optimizers/optimizer';
5454
export {RMSPropOptimizer} from './optimizers/rmsprop_optimizer';
5555
export {SGDOptimizer} from './optimizers/sgd_optimizer';
56-
export {Scalar, Tensor, Tensor1D, Tensor2D, Tensor3D, Tensor4D, TensorBuffer, variable, Variable} from './tensor';
56+
export {Scalar, Tensor, Tensor1D, Tensor2D, Tensor3D, Tensor4D, Tensor5D, TensorBuffer, variable, Variable} from './tensor';
5757
export {GradSaveFunc, NamedTensorMap, TensorContainer, TensorContainerArray, TensorContainerObject} from './tensor_types';
5858
export {DataType, DataTypeMap, DataValues, Rank, ShapeMap, TensorLike} from './types';
5959

0 commit comments

Comments
 (0)