본문 바로가기

ML 온디바이스

[Android] tensorflow lite 사용 시 GPU Delegate 지원 여부 확인

<CompatibilityList 클래스>

 

import org.tensorflow.lite.gpu.CompatibilityList;

 

CompatibilityList compatList = new CompatibilityList();

 

GPU 델리게이트가 추론을 실행하는 디바이스에서 지원하는지 확인

compatList.isDelegateSupportedOnThisDevice();

 

현재 디바이스에 맞는 옵션 추천

compatList.getBestOptionsForThisDevice();