From 45396b00099196d6d132c18dfbb9bbccac3d1c9e Mon Sep 17 00:00:00 2001 From: kangyizhang Date: Thu, 7 Nov 2019 09:53:40 -0800 Subject: [PATCH 1/2] add divnonan --- tfjs-node/src/nodejs_kernel_backend.ts | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/tfjs-node/src/nodejs_kernel_backend.ts b/tfjs-node/src/nodejs_kernel_backend.ts index c28fbcad46b..f08bf7a1f93 100644 --- a/tfjs-node/src/nodejs_kernel_backend.ts +++ b/tfjs-node/src/nodejs_kernel_backend.ts @@ -26,6 +26,7 @@ import {isArray, isNullOrUndefined} from 'util'; import {Int64Scalar} from './int64_tensors'; import {TensorMetadata, TFEOpAttr, TFJSBinding} from './tfjs_binding'; + type TensorData = { shape: number[], dtype: number, @@ -479,6 +480,12 @@ export class NodeJSKernelBackend extends KernelBackend { return this.executeSingleOutput('Div', opAttrs, [a, b]); } + divNoNan(a: Tensor, b: Tensor): Tensor { + const opAttrs = + [createTypeOpAttr('T', backend_util.upcastType(a.dtype, b.dtype))]; + return this.executeSingleOutput('DivNoNan', opAttrs, [a, b]); + } + unsortedSegmentSum( x: T, segmentIds: Tensor1D, numSegments: number): Tensor { const opAttrs = [ From 0ea2ac1db722b96cb6aa9ce0f39546e0f5ec09de Mon Sep 17 00:00:00 2001 From: kangyizhang Date: Thu, 7 Nov 2019 11:11:40 -0800 Subject: [PATCH 2/2] fix lint --- tfjs-node/src/nodejs_kernel_backend.ts | 1 - 1 file changed, 1 deletion(-) diff --git a/tfjs-node/src/nodejs_kernel_backend.ts b/tfjs-node/src/nodejs_kernel_backend.ts index f08bf7a1f93..5e4d9755749 100644 --- a/tfjs-node/src/nodejs_kernel_backend.ts +++ b/tfjs-node/src/nodejs_kernel_backend.ts @@ -26,7 +26,6 @@ import {isArray, isNullOrUndefined} from 'util'; import {Int64Scalar} from './int64_tensors'; import {TensorMetadata, TFEOpAttr, TFJSBinding} from './tfjs_binding'; - type TensorData = { shape: number[], dtype: number,