@tensorflow/tfjs-core
Version:
Hardware-accelerated JavaScript library for machine intelligence
6 lines (5 loc) • 676 B
TypeScript
import { NamedTensorMap } from '../tensor_types';
import { WeightsManifestConfig } from './types';
export declare function loadWeightsAsArrayBuffer(fetchURLs: string[], requestOptions?: RequestInit, fetchFunc?: Function, onProgress?: Function): Promise<ArrayBuffer[]>;
export declare function loadWeights(manifest: WeightsManifestConfig, filePathPrefix?: string, weightNames?: string[], requestOptions?: RequestInit): Promise<NamedTensorMap>;
export declare function weightsLoaderFactory(fetchWeightsFunction: (fetchUrls: string[]) => Promise<ArrayBuffer[]>): (manifest: WeightsManifestConfig, filePathPrefix?: string, weightNames?: string[]) => Promise<NamedTensorMap>;