onnxruntime/js/react_native/lib/index.ts

19 lines
624 B
TypeScript
Raw Normal View History

// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
export * from 'onnxruntime-common';
import {registerBackend, env} from 'onnxruntime-common';
import {Platform} from 'react-native';
import {onnxruntimeBackend} from './backend';
import {version} from './version';
registerBackend('cpu', onnxruntimeBackend, 1);
registerBackend('xnnpack', onnxruntimeBackend, 1);
if (Platform.OS === 'android') {
registerBackend('nnapi', onnxruntimeBackend, 1);
} else if (Platform.OS === 'ios') {
registerBackend('coreml', onnxruntimeBackend, 1);
}
env.versions['react-native'] = version;