From a53e45ff3d233de19df60e603a97808dcc4cb3db Mon Sep 17 00:00:00 2001 From: WenheLI Date: Wed, 24 Jul 2019 16:16:54 +0800 Subject: [PATCH 1/5] Add export for tfjs-node --- src/index.ts | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/src/index.ts b/src/index.ts index 15c34a75af..f8f8e72b41 100644 --- a/src/index.ts +++ b/src/index.ts @@ -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'; @@ -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. From d0a4d6b60910dc3e55a309395c888b69cf28a2fe Mon Sep 17 00:00:00 2001 From: WenheLI Date: Thu, 25 Jul 2019 23:22:10 +0800 Subject: [PATCH 2/5] remove convinfo --- src/index.ts | 1 - 1 file changed, 1 deletion(-) diff --git a/src/index.ts b/src/index.ts index f8f8e72b41..d72c6ddefa 100644 --- a/src/index.ts +++ b/src/index.ts @@ -58,7 +58,6 @@ export {GradSaveFunc, NamedTensorMap, TensorContainer, TensorContainerArray, Ten 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'; From 608defcc0bc6dc780f064b5ab315c34db5003333 Mon Sep 17 00:00:00 2001 From: WenheLI Date: Thu, 25 Jul 2019 23:38:10 +0800 Subject: [PATCH 3/5] fix some export in backend_util --- src/backends/backend_util.ts | 4 ++++ src/index.ts | 5 +---- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/src/backends/backend_util.ts b/src/backends/backend_util.ts index 34ab4f75e7..02f2e31c7e 100644 --- a/src/backends/backend_util.ts +++ b/src/backends/backend_util.ts @@ -23,11 +23,15 @@ import {hasEncodingLoss, makeZerosTypedArray} from '../util'; import {KernelBackend} from './backend'; +export {EPSILON_FLOAT32} from './backend'; + // Utilities needed by backend consumers of tf-core. export * from '../ops/axis_util'; export * from '../ops/broadcast_util'; export * from '../ops/concat_util'; export * from '../ops/conv_util'; +export {Activation} from '../ops/fused_util'; + export {TypedArray, upcastType, PixelData} from '../types'; export {MemoryInfo, TimingInfo} from '../engine'; diff --git a/src/index.ts b/src/index.ts index d72c6ddefa..d5615879c1 100644 --- a/src/index.ts +++ b/src/index.ts @@ -55,10 +55,9 @@ export {RMSPropOptimizer} from './optimizers/rmsprop_optimizer'; export {SGDOptimizer} from './optimizers/sgd_optimizer'; export {Scalar, Tensor, Tensor1D, Tensor2D, Tensor3D, Tensor4D, Tensor5D, TensorBuffer, variable, Variable} from './tensor'; export {GradSaveFunc, NamedTensorMap, TensorContainer, TensorContainerArray, TensorContainerObject} from './tensor_types'; -export {BackendValues, DataType, DataTypeMap, DataValues, Rank, ShapeMap, TensorLike, upcastType} from './types'; +export {BackendValues, DataType, DataTypeMap, DataValues, Rank, ShapeMap, TensorLike} from './types'; export * from './ops/ops'; -export {Activation} from './ops/fused_util'; export {LSTMCellFunc} from './ops/lstm'; export {Reduction} from './ops/loss_ops'; @@ -70,8 +69,6 @@ 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. From 86f907e54b87bd694280ae2ed9d19478b921ea39 Mon Sep 17 00:00:00 2001 From: WenheLI Date: Fri, 26 Jul 2019 20:16:00 +0800 Subject: [PATCH 4/5] Remove epsilon from backend_util --- src/backends/backend_util.ts | 2 -- 1 file changed, 2 deletions(-) diff --git a/src/backends/backend_util.ts b/src/backends/backend_util.ts index 02f2e31c7e..3e1f20ecdb 100644 --- a/src/backends/backend_util.ts +++ b/src/backends/backend_util.ts @@ -23,8 +23,6 @@ import {hasEncodingLoss, makeZerosTypedArray} from '../util'; import {KernelBackend} from './backend'; -export {EPSILON_FLOAT32} from './backend'; - // Utilities needed by backend consumers of tf-core. export * from '../ops/axis_util'; export * from '../ops/broadcast_util'; From c519b7eb6481b25e115a2782815fb4ef4650e5a8 Mon Sep 17 00:00:00 2001 From: WenheLI Date: Thu, 1 Aug 2019 21:42:14 +0800 Subject: [PATCH 5/5] Add backendvalues to backend_util --- src/backends/backend_util.ts | 2 +- src/index.ts | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/backends/backend_util.ts b/src/backends/backend_util.ts index 3e1f20ecdb..3de503109d 100644 --- a/src/backends/backend_util.ts +++ b/src/backends/backend_util.ts @@ -30,7 +30,7 @@ export * from '../ops/concat_util'; export * from '../ops/conv_util'; export {Activation} from '../ops/fused_util'; -export {TypedArray, upcastType, PixelData} from '../types'; +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 d5615879c1..453925df6f 100644 --- a/src/index.ts +++ b/src/index.ts @@ -55,7 +55,7 @@ export {RMSPropOptimizer} from './optimizers/rmsprop_optimizer'; export {SGDOptimizer} from './optimizers/sgd_optimizer'; export {Scalar, Tensor, Tensor1D, Tensor2D, Tensor3D, Tensor4D, Tensor5D, TensorBuffer, variable, Variable} from './tensor'; export {GradSaveFunc, NamedTensorMap, TensorContainer, TensorContainerArray, TensorContainerObject} from './tensor_types'; -export {BackendValues, DataType, DataTypeMap, DataValues, Rank, ShapeMap, TensorLike} from './types'; +export {DataType, DataTypeMap, DataValues, Rank, ShapeMap, TensorLike} from './types'; export * from './ops/ops'; export {LSTMCellFunc} from './ops/lstm';