Home
last modified time | relevance | path

Searched refs:TfLiteBackend (Results 1 – 10 of 10) sorted by relevance

/test/mlts/benchmark/src/com/android/nn/benchmark/core/
DTfLiteBackend.java19 public enum TfLiteBackend { enum
28 private TfLiteBackend(String name) { mName = name; } in TfLiteBackend() method in TfLiteBackend
30 public static TfLiteBackend parseString(String s) { in parseString()
31 for (TfLiteBackend backend : TfLiteBackend.values()) { in parseString()
DTestModels.java79 …public NNTestBase createNNTestBase(TfLiteBackend tfLiteBackend, boolean enableIntermediateTensorsD… in createNNTestBase()
85 TfLiteBackend tfLiteBackend = useNNAPI ? TfLiteBackend.NNAPI : TfLiteBackend.CPU; in createNNTestBase()
89 …public NNTestBase createNNTestBase(TfLiteBackend tfLiteBackend, boolean enableIntermediateTensorsD… in createNNTestBase()
DProcessor.java56 private TfLiteBackend mBackend;
85 mBackend = TfLiteBackend.CPU; in Processor()
89 setTfLiteBackend(useNNApi ? TfLiteBackend.NNAPI : TfLiteBackend.CPU); in setUseNNApi()
92 public void setTfLiteBackend(TfLiteBackend backend) { in setTfLiteBackend()
164 try (NNTestBase tb = testModelEntry.createNNTestBase(TfLiteBackend.NNAPI, in isTestModelSupportedByAccelerator()
193 if (mBackend == TfLiteBackend.NNAPI) { in changeTest()
DNNTestBase.java123 private TfLiteBackend mTfLiteBackend;
156 public void setTfLiteBackend(TfLiteBackend tfLiteBackend) { in setTfLiteBackend()
169 setTfLiteBackend(TfLiteBackend.NNAPI); in useNNApi()
173 if (mTfLiteBackend != TfLiteBackend.NNAPI) { in setNNApiDeviceName()
/test/mlts/benchmark/src/com/android/nn/benchmark/app/
DNNBenchmark.java30 import com.android.nn.benchmark.core.TfLiteBackend;
63 mProcessor.setTfLiteBackend(useNNApi ? TfLiteBackend.NNAPI : TfLiteBackend.CPU); in setUseNNApi()
132 …teBackend(!i.getBooleanExtra(EXTRA_DISABLE_NNAPI, false) ? TfLiteBackend.NNAPI : TfLiteBackend.CPU… in onResume()
/test/mlts/benchmark/src/com/android/nn/benchmark/util/
DDumpIntermediateTensors.java26 import com.android.nn.benchmark.core.TfLiteBackend;
94 TfLiteBackend backend = useNNAPI ? TfLiteBackend.NNAPI : TfLiteBackend.CPU; in onCreate()
/test/mlts/benchmark/src/com/android/nn/crashtest/core/test/
DPerformanceDegradationTest.java28 import com.android.nn.benchmark.core.TfLiteBackend;
230 benchmarkProcessor.setTfLiteBackend(TfLiteBackend.NNAPI); in modelPerformanceCollector()
256 mProcessor.setTfLiteBackend(TfLiteBackend.NNAPI);
DRunModelsInParallel.java29 import com.android.nn.benchmark.core.TfLiteBackend;
135 result.setTfLiteBackend(TfLiteBackend.NNAPI); in createSubTestRunner()
/test/mlts/benchmark/src/com/android/nn/crashtest/app/
DAcceleratorSpecificTestSupport.java30 import com.android.nn.benchmark.core.TfLiteBackend;
126 mProcessor.setTfLiteBackend(TfLiteBackend.NNAPI); in DriverLivenessChecker()
/test/mlts/benchmark/dogfood/src/com/android/nn/dogfood/
DBenchmarkJobService.java33 import com.android.nn.benchmark.core.TfLiteBackend;
81 mProcessor.setTfLiteBackend(TfLiteBackend.NNAPI); in doBenchmark()