Searched refs:computeMode (Results 1 – 7 of 7) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/runtime/test/ |
| H A D | TestMain.cpp | 54 static int test(bool useCpuOnly, Execution::ComputeMode computeMode) { in test() argument
60 uint32_t passIndex = (useCpuOnly << 0) + (static_cast<uint32_t>(computeMode) << 1); in test()
70 Execution::setComputeMode(computeMode); in test()
72 auto computeModeText = [computeMode] { in test()
73 switch (computeMode) { in test()
|
| H A D | TestUnknownDimensions.cpp | 89 Execution::ComputeMode computeMode);
170 Execution::ComputeMode computeMode) { in TestOne() argument
301 ASSERT_EQ(execution.compute(computeMode), Result::NO_ERROR); in TestOne()
332 for (auto computeMode : computeModes) { in TestAll() local
334 paramsForOutput, computeMode); in TestAll()
|
| H A D | SupportLibraryTestGenerated.cpp | 71 Execution* execution, Execution::ComputeMode computeMode,
245 Execution::ComputeMode computeMode, Result* result, in computeWithPtrs() argument
251 *result = execution->compute(computeMode); in computeWithPtrs()
285 Execution::ComputeMode computeMode, Result* result, in computeWithDeviceMemories() argument
328 *result = execution->compute(computeMode); in computeWithDeviceMemories()
|
| H A D | TestPartitioningRandom.cpp | 696 auto computeMode = WrapperExecution::getComputeMode(); in TEST_P() local
701 if (randFrac() < 0.5 && computeMode == WrapperExecution::ComputeMode::ASYNC) { in TEST_P()
702 computeMode = WrapperExecution::ComputeMode::FENCED; in TEST_P()
1032 if (computeMode == WrapperExecution::ComputeMode::FENCED) { in TEST_P()
1044 computeMode = WrapperExecution::ComputeMode::ASYNC; in TEST_P()
1378 ASSERT_EQ(e.compute(computeMode), Result::NO_ERROR); in TEST_P()
1411 ASSERT_EQ(e2.compute(computeMode), Result::NO_ERROR); in TEST_P()
|
| H A D | TestNeuralNetworksWrapper.h | 489 Result compute(ComputeMode computeMode = mComputeMode) {
490 switch (computeMode) {
|
| /aosp12/packages/modules/NeuralNetworks/ |
| H A D | TEST_MAPPING | 23 // * computeMode = ComputeMode::ASYNC
|
| /aosp12/packages/modules/NeuralNetworks/shim_and_sl/include/ |
| H A D | SupportLibraryWrapper.h | 592 Result compute(ComputeMode computeMode = mComputeMode) {
593 switch (computeMode) {
|