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.java83 …public NNTestBase createNNTestBase(TfLiteBackend tfLiteBackend, boolean enableIntermediateTensorsD… in createNNTestBase()
90 TfLiteBackend tfLiteBackend = useNNAPI ? TfLiteBackend.NNAPI : TfLiteBackend.CPU; in createNNTestBase()
95 …public NNTestBase createNNTestBase(TfLiteBackend tfLiteBackend, boolean enableIntermediateTensorsD… in createNNTestBase()
DProcessor.java57 private TfLiteBackend mBackend;
94 mBackend = TfLiteBackend.CPU; in Processor()
102 setTfLiteBackend(useNNApi ? TfLiteBackend.NNAPI : TfLiteBackend.CPU); in setUseNNApi()
105 public void setTfLiteBackend(TfLiteBackend backend) { in setTfLiteBackend()
185 try (NNTestBase tb = testModelEntry.createNNTestBase(TfLiteBackend.NNAPI, in isTestModelSupportedByAccelerator()
218 if (mBackend == TfLiteBackend.NNAPI) { in changeTest()
DNNTestBase.java143 private TfLiteBackend mTfLiteBackend;
205 public void setTfLiteBackend(TfLiteBackend tfLiteBackend) { in setTfLiteBackend()
218 setTfLiteBackend(TfLiteBackend.NNAPI); in useNNApi()
226 if (mTfLiteBackend != TfLiteBackend.NNAPI) { in setNNApiDeviceName()
/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/benchmark/app/
DNNBenchmark.java30 import com.android.nn.benchmark.core.TfLiteBackend;
67 mProcessor.setTfLiteBackend(useNNApi ? TfLiteBackend.NNAPI : TfLiteBackend.CPU); in setUseNNApi()
151 …teBackend(!i.getBooleanExtra(EXTRA_DISABLE_NNAPI, false) ? TfLiteBackend.NNAPI : TfLiteBackend.CPU… in onResume()
DAcceleratorSpecificTestSupport.java30 import com.android.nn.benchmark.core.TfLiteBackend;
164 mProcessor.setTfLiteBackend(TfLiteBackend.NNAPI);
/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()
/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;
146 result.setTfLiteBackend(TfLiteBackend.NNAPI); in createSubTestRunner()