Lines Matching refs:compilation

68     void executeWithCompilation(const Compilation& compilation, const TestModel& testModel);
129 Compilation compilation(&model); in compileModel() local
130 Result result = compilation.finish(); in compileModel()
137 return compilation; in compileModel()
150 static ANeuralNetworksMemory* createDeviceMemoryForInput(const Compilation& compilation, in createDeviceMemoryForInput() argument
154 EXPECT_EQ(ANeuralNetworksMemoryDesc_addInputRole(desc, compilation.getHandle(), index, 1.0f), in createDeviceMemoryForInput()
163 static ANeuralNetworksMemory* createDeviceMemoryForOutput(const Compilation& compilation, in createDeviceMemoryForOutput() argument
167 EXPECT_EQ(ANeuralNetworksMemoryDesc_addOutputRole(desc, compilation.getHandle(), index, 1.0f), in createDeviceMemoryForOutput()
177 static void computeWithDeviceMemories(const Compilation& compilation, const TestModel& testModel, in computeWithDeviceMemories() argument
199 ANeuralNetworksMemory* memory = createDeviceMemoryForInput(compilation, i); in computeWithDeviceMemories()
214 ANeuralNetworksMemory* memory = createDeviceMemoryForOutput(compilation, i); in computeWithDeviceMemories()
239 void GeneratedTests::executeWithCompilation(const Compilation& compilation, in executeWithCompilation() argument
243 Execution execution(&compilation); in executeWithCompilation()
248 computeWithDeviceMemories(compilation, testModel, &execution, &result, &outputs); in executeWithCompilation()
281 std::optional<Compilation> compilation = compileModel(model); in executeOnce() local
283 if (!compilation) return; in executeOnce()
284 executeWithCompilation(compilation.value(), testModel); in executeOnce()
302 std::optional<Compilation> compilation = compileModel(model); in executeMultithreadedSharedCompilation() local
304 if (!compilation) return; in executeMultithreadedSharedCompilation()
308 std::thread([&]() { executeWithCompilation(compilation.value(), testModel); })); in executeMultithreadedSharedCompilation()