UNPKG

3.46 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 concatenateArrayBuffers: typeof import("@tensorflow/tfjs-core/dist/io/io_utils").concatenateArrayBuffers;
40 decodeWeights: typeof import("@tensorflow/tfjs-core/dist/io/io_utils").decodeWeights;
41 encodeWeights: typeof import("@tensorflow/tfjs-core/dist/io/io_utils").encodeWeights;
42 fromMemory: typeof import("@tensorflow/tfjs-core/dist/io/passthrough").fromMemory;
43 fromMemorySync: typeof import("@tensorflow/tfjs-core/dist/io/passthrough").fromMemorySync;
44 getLoadHandlers: (url: string | string[], loadOptions?: tf.io.LoadOptions) => tf.io.IOHandler[];
45 getModelArtifactsForJSON: typeof import("@tensorflow/tfjs-core/dist/io/io_utils").getModelArtifactsForJSON;
46 getModelArtifactsInfoForJSON: typeof import("@tensorflow/tfjs-core/dist/io/io_utils").getModelArtifactsInfoForJSON;
47 getSaveHandlers: (url: string | string[]) => tf.io.IOHandler[];
48 http: typeof import("@tensorflow/tfjs-core/dist/io/http").http;
49 isHTTPScheme: typeof import("@tensorflow/tfjs-core/dist/io/http").isHTTPScheme;
50 loadWeights: typeof import("@tensorflow/tfjs-core/dist/io/weights_loader").loadWeights;
51 registerLoadRouter: (loudRouter: import("@tensorflow/tfjs-core/dist/io/router_registry").IORouter) => void;
52 registerSaveRouter: (loudRouter: import("@tensorflow/tfjs-core/dist/io/router_registry").IORouter) => void;
53 weightsLoaderFactory: typeof import("@tensorflow/tfjs-core/dist/io/weights_loader").weightsLoaderFactory;
54 withSaveHandler: typeof import("@tensorflow/tfjs-core/dist/io/passthrough").withSaveHandler;
55 withSaveHandlerSync: typeof import("@tensorflow/tfjs-core/dist/io/passthrough").withSaveHandlerSync;
56};
57export * from '@tensorflow/tfjs';
58export * from './node';