瀏覽代碼

Update tf.js

Lutz Roeder 3 年之前
父節點
當前提交
f9eb24bfdf
共有 1 個文件被更改,包括 4 次插入3 次删除
  1. 4 3
      source/tf.js

+ 4 - 3
source/tf.js

@@ -587,7 +587,7 @@ tf.ModelFactory = class {
                     return openBinarySavedModel({ stream: stream });
                 });
             };
-            const openMemmappedFileSystemDirectory = (context) => {
+            const openMemmapped = (context) => {
                 const stream = context.stream;
                 const readDirectoryOffset = (stream) => {
                     stream.seek(-8);
@@ -671,7 +671,7 @@ tf.ModelFactory = class {
                 case 'tf.pb.keras.SavedMetadata':
                     return openSavedMetadata(context);
                 case 'tf.pb.mmap':
-                    return openMemmappedFileSystemDirectory(context);
+                    return openMemmapped(context);
                 default:
                     throw new tf.Error("Unsupported TensorFlow format '" + match + "'.");
             }
@@ -1267,8 +1267,9 @@ tf.Tensor = class {
                         this._data = tensor.double_val || null;
                         break;
                     }
-                    case DataType.DT_INT8:
                     case DataType.DT_UINT8:
+                    case DataType.DT_UINT16:
+                    case DataType.DT_INT8:
                     case DataType.DT_INT16:
                     case DataType.DT_INT32: {
                         this._data = tensor.int_val || null;