Lutz Roeder 3 лет назад
Родитель
Сommit
7a8c48f6d6
1 измененных файлов с 1 добавлено и 0 удалено
  1. 1 0
      source/tf.js

+ 1 - 0
source/tf.js

@@ -1254,6 +1254,7 @@ tf.Tensor = class {
                     }
                     case DataType.DT_INT8:
                     case DataType.DT_UINT8:
+                    case DataType.DT_INT16:
                     case DataType.DT_INT32: {
                         this._data = tensor.int_val || null;
                         break;