UNPKG

react-native-fast-tflite

Version:

High-performance TensorFlow Lite library for React Native

20 lines (16 loc) 877 B
import { ConfigPlugin, createRunOncePlugin } from '@expo/config-plugins' import { ConfigProps } from './@types' import { withCoreMLDelegate } from './withCoreMLDelegate' import { withAndroidGpuLibraries } from './withAndroidGpuLibraries' // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment, @typescript-eslint/no-var-requires const pkg = require('../../../package.json') // from the lib directory, the package.json is three levels up const withFastTFLite: ConfigPlugin<ConfigProps> = (config, props) => { if (props?.enableCoreMLDelegate) config = withCoreMLDelegate(config) if ( props?.enableAndroidGpuLibraries === true || Array.isArray(props?.enableAndroidGpuLibraries) ) config = withAndroidGpuLibraries(config, props.enableAndroidGpuLibraries) return config } export default createRunOncePlugin(withFastTFLite, pkg.name, pkg.version)