Ver código fonte

Fix TensorFlow metadata (#840)

Lutz Roeder 4 anos atrás
pai
commit
5198db554e
1 arquivos alterados com 1 adições e 0 exclusões
  1. 1 0
      source/tf.js

+ 1 - 0
source/tf.js

@@ -1983,6 +1983,7 @@ tf.GraphMetadata = class {
             this._functions.set(name, new tf.Function(this, name, null));
             return this._functions.get(name);
         }
+        return type;
     }
 
     attribute(type, name) {