Fix indexeddb for 1GB models (#7725)

* Fix indexeddb for 1GB models

Fixes #7702 by concatenating model weights into a single ArrayBuffer before
sending them to IndexedDB. A better solution would be to store the model as
multiple records, but this quick fix is easy to implement solves the issue for
most current models (~1GB).

* Use CompositeArrayBuffer.join

tfjs-core/src/io/indexed_db.ts

@@ -22,6 +22,7 @@ import {env} from '../environment';
 import {getModelArtifactsInfoForJSON} from './io_utils';
 import {IORouter, IORouterRegistry} from './router_registry';
 import {IOHandler, ModelArtifacts, ModelArtifactsInfo, ModelStoreManager, SaveResult} from './types';
+import {CompositeArrayBuffer} from './composite_array_buffer';
 
 const DATABASE_NAME = 'tensorflowjs';
 const DATABASE_VERSION = 1;
@@ -157,6 +158,13 @@ export class BrowserIndexedDB implements IOHandler {
           modelTx.oncomplete = () => db.close();
         } else {
           // Put model into object store.
+
+          // Concatenate all the model weights into a single ArrayBuffer. Large
+          // models (~1GB) have problems saving if they are not concatenated.
+          // TODO(mattSoulanille): Save large models to multiple indexeddb
+          // records.
+          modelArtifacts.weightData = CompositeArrayBuffer.join(
+              modelArtifacts.weightData);
           const modelArtifactsInfo: ModelArtifactsInfo =
               getModelArtifactsInfoForJSON(modelArtifacts);
           // First, put ModelArtifactsInfo into info store.