human/src/tfjs/tf-browser.ts

43 lines
1.7 KiB
TypeScript
Raw Normal View History

2020-11-17 16:18:15 +01:00
// wrapper to load tfjs in a single place so version can be changed quickly
2020-11-18 14:26:28 +01:00
// simplified
// { modules: 1061, moduleBytes: 3772720, outputBytes: 1531035 }
2020-12-13 00:34:30 +01:00
// export * from '@tensorflow/tfjs/dist/index.js';
// export * from '@tensorflow/tfjs-backend-wasm';
2020-11-18 14:26:28 +01:00
// modular
// { modules: 1064, moduleBytes: 3793219, outputBytes: 1535600 }
// get versions of all packages.
import { version as tfjs } from '@tensorflow/tfjs/package.json';
import { version as versionCore } from '@tensorflow/tfjs-core/package.json';
import { version as versionData } from '@tensorflow/tfjs-data/package.json';
import { version as versionLayers } from '@tensorflow/tfjs-layers/package.json';
import { version as versionConverter } from '@tensorflow/tfjs-converter/package.json';
// for backends, get version from source so it can register backend during import
import { version_cpu } from '@tensorflow/tfjs-backend-cpu/dist/index.js';
import { version_webgl } from '@tensorflow/tfjs-backend-webgl/dist/index.js';
import { version_wasm } from '@tensorflow/tfjs-backend-wasm/dist/index.js';
// export all
export * from '@tensorflow/tfjs-core/dist/index.js';
export * from '@tensorflow/tfjs-layers/dist/index.js';
export * from '@tensorflow/tfjs-converter/dist/index.js';
export * as data from '@tensorflow/tfjs-data/dist/index.js';
export * from '@tensorflow/tfjs-backend-cpu/dist/index.js';
export * from '@tensorflow/tfjs-backend-webgl/dist/index.js';
export * from '@tensorflow/tfjs-backend-wasm/dist/index.js';
// export versions
export const version = {
tfjs,
'tfjs-core': versionCore,
'tfjs-data': versionData,
'tfjs-layers': versionLayers,
'tfjs-converter': versionConverter,
'tfjs-backend-cpu': version_cpu,
'tfjs-backend-webgl': version_webgl,
'tfjs-backend-wasm': version_wasm,
};