UNPKG

3.14 kBTypeScriptView Raw
1/**
2 * @license
3 * Copyright 2018 Google Inc. 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 * as tf from '@tensorflow/tfjs';
18import * as nodeIo from './io/index';
19import './kernels/all_kernels';
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 getLoadHandlers: (url: string | string[], loadOptions?: tf.io.LoadOptions) => tf.io.IOHandler[];
44 getModelArtifactsInfoForJSON: typeof import("@tensorflow/tfjs-core/dist/io/io_utils").getModelArtifactsInfoForJSON;
45 getSaveHandlers: (url: string | string[]) => tf.io.IOHandler[];
46 http: typeof import("@tensorflow/tfjs-core/dist/io/http").http;
47 isHTTPScheme: typeof import("@tensorflow/tfjs-core/dist/io/http").isHTTPScheme;
48 loadWeights: typeof import("@tensorflow/tfjs-core/dist/io/weights_loader").loadWeights;
49 registerLoadRouter: (loudRouter: import("@tensorflow/tfjs-core/dist/io/router_registry").IORouter) => void;
50 registerSaveRouter: (loudRouter: import("@tensorflow/tfjs-core/dist/io/router_registry").IORouter) => void;
51 weightsLoaderFactory: typeof import("@tensorflow/tfjs-core/dist/io/weights_loader").weightsLoaderFactory;
52 withSaveHandler: typeof import("@tensorflow/tfjs-core/dist/io/passthrough").withSaveHandler;
53};
54export * from '@tensorflow/tfjs';
55export * from './node';