@tensorflow/tfjs-converter - npm Package Compare versions

Comparing version 2.4.0 to 2.5.0

dist/executor/graph_executor.js

@@ -151,3 +151,4 @@ /**
         const inputNodes = names.map(name => this.graph.nodes[parseNodeName(name)[0]]);
-        const outputNodes = outputs.map(name => this.graph.nodes[parseNodeName(name)[0]]);
+        const outputNodeNames = outputs.map(name => parseNodeName(name)[0]);
+        const outputNodes = outputNodeNames.map(name => this.graph.nodes[name]);
         const compilationKey = this.getCompilationKey(inputNodes, outputNodes);
@@ -182,3 +183,3 @@ // Do nothing if the compiled graph cache contains the input.
                     tensorsMap[node.name] = tensors;
-                    this.checkTensorForDisposal(node.name, node, tensorsMap, context, tensorsToKeep, outputs, intermediateTensorConsumerCount);
+                    this.checkTensorForDisposal(node.name, node, tensorsMap, context, tensorsToKeep, outputNodeNames, intermediateTensorConsumerCount);
                 }
@@ -311,3 +312,4 @@ }
         const inputNodes = names.map(name => this.graph.nodes[parseNodeName(name)[0]]);
-        const outputNodes = outputNames.map(name => this.graph.nodes[parseNodeName(name)[0]]);
+        const outputNodeNames = outputNames.map(name => parseNodeName(name)[0]);
+        const outputNodes = outputNodeNames.map(name => this.graph.nodes[name]);
         const { usedNodes, missingInputs, dynamicNode, syncInputs } = getExecutionSubgraph(inputs, outputNodes, this.weightMap);
@@ -328,3 +330,3 @@ const stack = [...inputNodes, ...this.graph.weights].map(node => {
         while (stack.length > 0) {
-            const promises = this.processStack(inputNodes, stack, context, tensorsMap, added, tensorsToKeep, outputNames, intermediateTensorConsumerCount, usedNodes);
+            const promises = this.processStack(inputNodes, stack, context, tensorsMap, added, tensorsToKeep, outputNodeNames, intermediateTensorConsumerCount, usedNodes);
             await Promise.all(promises);
@@ -331,0 +333,0 @@ }

dist/version.d.ts

 /** @license See the LICENSE file. */
-declare const version = "2.4.0";
+declare const version = "2.5.0";
 export { version };

dist/version.js

 /** @license See the LICENSE file. */
 // This code is auto-generated, do not modify this file!
-const version = '2.4.0';
+const version = '2.5.0';
 export { version };
 //# sourceMappingURL=version.js.map

package.json

 {
   "name": "@tensorflow/tfjs-converter",
-  "version": "2.4.0",
+  "version": "2.5.0",
   "description": "Tensorflow model converter for javascript",
@@ -18,3 +18,3 @@ "main": "dist/tf-converter.node.js",
   "peerDependencies": {
-    "@tensorflow/tfjs-core": "2.4.0"
+    "@tensorflow/tfjs-core": "2.5.0"
   },
@@ -26,4 +26,4 @@ "devDependencies": {
     "@rollup/plugin-typescript": "^3.0.0",
-    "@tensorflow/tfjs-backend-cpu": "2.4.0",
-    "@tensorflow/tfjs-core": "2.4.0",
+    "@tensorflow/tfjs-backend-cpu": "2.5.0",
+    "@tensorflow/tfjs-core": "2.5.0",
     "@types/argparse": "^1.0.38",
@@ -49,3 +49,3 @@ "@types/deep-equal": "^1.0.1",
     "karma-typescript": "~4.0.0",
-    "node-fetch": "~2.1.2",
+    "node-fetch": "~2.6.1",
     "opn": "~5.1.0",
@@ -52,0 +52,0 @@ "protobufjs": "~6.8.6",

README.md

@@ -51,4 +51,5 @@ # Getting started
 
+Install the library with interactive CLI:
 ```bash
- pip install tensorflowjs
+ pip install tensorflowjs[wizard]
 ```
@@ -55,0 +56,0 @@

src/executor/graph_executor.ts

@@ -186,4 +186,4 @@ /**
         names.map(name => this.graph.nodes[parseNodeName(name)[0]]);
-    const outputNodes =
-        outputs.map(name => this.graph.nodes[parseNodeName(name)[0]]);
+    const outputNodeNames = outputs.map(name => parseNodeName(name)[0]);
+    const outputNodes = outputNodeNames.map(name => this.graph.nodes[name]);
     const compilationKey = this.getCompilationKey(inputNodes, outputNodes);
@@ -222,4 +222,4 @@ // Do nothing if the compiled graph cache contains the input.
           this.checkTensorForDisposal(
-              node.name, node, tensorsMap, context, tensorsToKeep, outputs,
-              intermediateTensorConsumerCount);
+              node.name, node, tensorsMap, context, tensorsToKeep,
+              outputNodeNames, intermediateTensorConsumerCount);
         }
@@ -382,4 +382,4 @@ }
         names.map(name => this.graph.nodes[parseNodeName(name)[0]]);
-    const outputNodes =
-        outputNames.map(name => this.graph.nodes[parseNodeName(name)[0]]);
+    const outputNodeNames = outputNames.map(name => parseNodeName(name)[0]);
+    const outputNodes = outputNodeNames.map(name => this.graph.nodes[name]);
     const {usedNodes, missingInputs, dynamicNode, syncInputs} =
@@ -405,3 +405,3 @@ getExecutionSubgraph(inputs, outputNodes, this.weightMap);
           inputNodes, stack, context, tensorsMap, added, tensorsToKeep,
-          outputNames, intermediateTensorConsumerCount, usedNodes);
+          outputNodeNames, intermediateTensorConsumerCount, usedNodes);
       await Promise.all(promises);
@@ -408,0 +408,0 @@ }

src/version.ts

 /** @license See the LICENSE file. */
 
 // This code is auto-generated, do not modify this file!
-const version = '2.4.0';
+const version = '2.5.0';
 export {version};

No alert changes

Improved metrics

Total package byte prevSize

5928350

increased by0.06%

Lines of code

43188

increased by0.02%

Number of lines in readme file

505

increased by0.2%

No dependency changes