Lines Matching refs:compilation
76 ANeuralNetworksMemory* createDeviceMemoryForInput(const Compilation& compilation,
78 ANeuralNetworksMemory* createDeviceMemoryForOutput(const Compilation& compilation,
81 const Compilation& compilation, const TestModel& testModel, Execution* execution,
85 void executeWithCompilation(const Compilation& compilation, const TestModel& testModel);
241 auto compilation = std::move(resultCompilation.second); in compileModel() local
242 Result result = compilation.finish(); in compileModel()
249 return compilation; in compileModel()
263 ANeuralNetworksMemory* GeneratedTests::createDeviceMemoryForInput(const Compilation& compilation, in createDeviceMemoryForInput() argument
268 desc, compilation.getHandle(), index, 1.0f), in createDeviceMemoryForInput()
277 ANeuralNetworksMemory* GeneratedTests::createDeviceMemoryForOutput(const Compilation& compilation, in createDeviceMemoryForOutput() argument
282 desc, compilation.getHandle(), index, 1.0f), in createDeviceMemoryForOutput()
293 const Compilation& compilation, const TestModel& testModel, Execution* execution, in computeWithDeviceMemories() argument
314 ANeuralNetworksMemory* memory = createDeviceMemoryForInput(compilation, i); in computeWithDeviceMemories()
331 ANeuralNetworksMemory* memory = createDeviceMemoryForOutput(compilation, i); in computeWithDeviceMemories()
360 void GeneratedTests::executeWithCompilation(const Compilation& compilation, in executeWithCompilation() argument
364 Execution execution(mNnApi.get(), &compilation); in executeWithCompilation()
369 if (computeWithDeviceMemories(compilation, testModel, &execution, mComputeMode, &result, in executeWithCompilation()
424 std::optional<Compilation> compilation = compileModel(model, device); in executeOnce() local
427 if (!compilation) return; in executeOnce()
428 executeWithCompilation(compilation.value(), testModel); in executeOnce()
471 std::optional<Compilation> compilation = compileModel(model, device); in executeMultithreadedSharedCompilation() local
474 if (!compilation) return; in executeMultithreadedSharedCompilation()
478 std::thread([&]() { executeWithCompilation(compilation.value(), testModel); })); in executeMultithreadedSharedCompilation()