react-native-fast-tflite
Version:
High-performance TensorFlow Lite library for React Native
20 lines (16 loc) • 877 B
text/typescript
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)