UNPKG

3.89 kBTypeScriptView Raw
1/**
2 * @license
3 * Copyright 2018 Google LLC. All Rights Reserved.
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 * =============================================================================
16 */
17import './register_all_kernels';
18import * as tf from '@tensorflow/tfjs';
19import * as nodeIo from './io/index';
20export declare const version: {
21 'tfjs-node': string;
22 'tfjs-core': string;
23 'tfjs-backend-cpu': string;
24 'tfjs-backend-webgl': string;
25 'tfjs-data': string;
26 'tfjs-layers': string;
27 'tfjs-converter': string;
28 tfjs: string;
29};
30export declare const io: {
31 fileSystem: typeof nodeIo.fileSystem;
32 nodeHTTPRequest: typeof nodeIo.nodeHTTPRequest;
33 copyModel: typeof import("@tensorflow/tfjs-core/dist/io/model_management").copyModel;
34 listModels: typeof import("@tensorflow/tfjs-core/dist/io/model_management").listModels;
35 moveModel: typeof import("@tensorflow/tfjs-core/dist/io/model_management").moveModel;
36 removeModel: typeof import("@tensorflow/tfjs-core/dist/io/model_management").removeModel;
37 browserFiles: typeof import("@tensorflow/tfjs-core/dist/io/browser_files").browserFiles;
38 browserHTTPRequest: typeof import("@tensorflow/tfjs-core/dist/io/http").browserHTTPRequest;
39 CompositeArrayBuffer: typeof import("@tensorflow/tfjs-core/dist/io/composite_array_buffer").CompositeArrayBuffer;
40 concatenateArrayBuffers: typeof import("@tensorflow/tfjs-core/dist/io/io_utils").concatenateArrayBuffers;
41 decodeWeights: typeof import("@tensorflow/tfjs-core/dist/io/io_utils").decodeWeights;
42 decodeWeightsStream: typeof import("@tensorflow/tfjs-core/dist/io/io_utils").decodeWeightsStream;
43 encodeWeights: typeof import("@tensorflow/tfjs-core/dist/io/io_utils").encodeWeights;
44 fromMemory: typeof import("@tensorflow/tfjs-core/dist/io/passthrough").fromMemory;
45 fromMemorySync: typeof import("@tensorflow/tfjs-core/dist/io/passthrough").fromMemorySync;
46 getLoadHandlers: (url: string | string[], loadOptions?: tf.io.LoadOptions) => tf.io.IOHandler[];
47 getModelArtifactsForJSON: typeof import("@tensorflow/tfjs-core/dist/io/io_utils").getModelArtifactsForJSON;
48 getModelArtifactsForJSONSync: typeof import("@tensorflow/tfjs-core/dist/io/io_utils").getModelArtifactsForJSONSync;
49 getModelArtifactsInfoForJSON: typeof import("@tensorflow/tfjs-core/dist/io/io_utils").getModelArtifactsInfoForJSON;
50 getSaveHandlers: (url: string | string[]) => tf.io.IOHandler[];
51 getWeightSpecs: typeof import("@tensorflow/tfjs-core/dist/io/io_utils").getWeightSpecs;
52 http: typeof import("@tensorflow/tfjs-core/dist/io/http").http;
53 isHTTPScheme: typeof import("@tensorflow/tfjs-core/dist/io/http").isHTTPScheme;
54 loadWeights: typeof import("@tensorflow/tfjs-core/dist/io/weights_loader").loadWeights;
55 registerLoadRouter: (loudRouter: import("@tensorflow/tfjs-core/dist/io/router_registry").IORouter) => void;
56 registerSaveRouter: (loudRouter: import("@tensorflow/tfjs-core/dist/io/router_registry").IORouter) => void;
57 weightsLoaderFactory: typeof import("@tensorflow/tfjs-core/dist/io/weights_loader").weightsLoaderFactory;
58 withSaveHandler: typeof import("@tensorflow/tfjs-core/dist/io/passthrough").withSaveHandler;
59 withSaveHandlerSync: typeof import("@tensorflow/tfjs-core/dist/io/passthrough").withSaveHandlerSync;
60};
61export * from '@tensorflow/tfjs';
62export * from './node';