Searched refs:computeMode (Results 1 – 7 of 7) sorted by relevance
| /packages/modules/NeuralNetworks/runtime/test/ |
| 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()
|
| D | TestUnknownDimensions.cpp | 90 Execution::ComputeMode computeMode);
171 Execution::ComputeMode computeMode) { in TestOne() argument
302 ASSERT_EQ(execution.compute(computeMode), Result::NO_ERROR); in TestOne()
333 for (auto computeMode : computeModes) { in TestAll() local
335 paramsForOutput, computeMode); in TestAll()
|
| D | SupportLibraryTestGenerated.cpp | 82 Execution::ComputeMode computeMode, Result* result, std::vector<TestBuffer>* outputs);
254 Execution::ComputeMode computeMode, Result* result, in computeWithPtrs() argument
260 *result = execution->compute(computeMode); in computeWithPtrs()
294 Execution::ComputeMode computeMode, Result* result, std::vector<TestBuffer>* outputs) { in computeWithDeviceMemories() argument
340 *result = execution->compute(computeMode); in computeWithDeviceMemories()
|
| D | TestPartitioningRandom.cpp | 679 auto computeMode = WrapperExecution::getComputeMode(); in TEST_P() local
684 if (randFrac() < 0.5 && computeMode == WrapperExecution::ComputeMode::ASYNC) { in TEST_P()
685 computeMode = WrapperExecution::ComputeMode::FENCED; in TEST_P()
1015 if (computeMode == WrapperExecution::ComputeMode::FENCED) { in TEST_P()
1027 computeMode = WrapperExecution::ComputeMode::ASYNC; in TEST_P()
1362 ASSERT_EQ(e.compute(computeMode), Result::NO_ERROR); in TEST_P()
1395 ASSERT_EQ(e2.compute(computeMode), Result::NO_ERROR); in TEST_P()
|
| D | TestNeuralNetworksWrapper.h | 491 Result compute(ComputeMode computeMode = mComputeMode) {
492 switch (computeMode) {
|
| /packages/modules/NeuralNetworks/ |
| D | TEST_MAPPING | 28 // * computeMode = ComputeMode::ASYNC
45 // // * computeMode = ComputeMode::ASYNC
|
| /packages/modules/NeuralNetworks/shim_and_sl/include/ |
| D | SupportLibraryWrapper.h | 607 Result compute(ComputeMode computeMode = mComputeMode) {
608 switch (computeMode) {
|