Lines Matching refs:compilation

368     void executeWithMemoryAsInput(ANeuralNetworksCompilation* compilation,  in executeWithMemoryAsInput()  argument
372 ASSERT_EQ(ANeuralNetworksExecution_create(compilation, &execution), in executeWithMemoryAsInput()
382 void executeWithMemoryAsOutput(ANeuralNetworksCompilation* compilation, in executeWithMemoryAsOutput() argument
386 ASSERT_EQ(ANeuralNetworksExecution_create(compilation, &execution), in executeWithMemoryAsOutput()
883 ANeuralNetworksCompilation* compilation = nullptr; in TEST_F() local
884 EXPECT_EQ(ANeuralNetworksCompilation_create(nullptr, &compilation), in TEST_F()
887 EXPECT_EQ(ANeuralNetworksCompilation_create(mModel, &compilation), ANEURALNETWORKS_BAD_STATE); in TEST_F()
898 ANeuralNetworksCompilation* compilation = nullptr; in TEST_F() local
899 EXPECT_EQ(ANeuralNetworksCompilation_createForDevices(nullptr, &device, 1, &compilation), in TEST_F()
905 EXPECT_EQ(ANeuralNetworksCompilation_createForDevices(mModel, &device, 0, &compilation), in TEST_F()
911 &compilation), in TEST_F()
916 &compilation), in TEST_F()
920 ANeuralNetworksCompilation* compilation = nullptr; in TEST_F() local
921 EXPECT_EQ(ANeuralNetworksCompilation_createForDevices(nullptr, nullptr, 1, &compilation), in TEST_F()
925 EXPECT_EQ(ANeuralNetworksCompilation_createForDevices(mModel, nullptr, 1, &compilation), in TEST_F()
1466 static void testConcurrentExecution(bool reusable, ANeuralNetworksCompilation* compilation) { in testConcurrentExecution() argument
1467 ASSERT_EQ(ANeuralNetworksCompilation_finish(compilation), ANEURALNETWORKS_NO_ERROR); in testConcurrentExecution()
1470 const auto compute = [compilation](ExecutionType executionType, in testConcurrentExecution()
1487 int result = ANeuralNetworksBurst_create(compilation, &burst); in testConcurrentExecution()
1515 ASSERT_EQ(ANeuralNetworksExecution_create(compilation, &execution), in testConcurrentExecution()
2583 ANeuralNetworksCompilation* compilation; in TEST_F() local
2584 ASSERT_EQ(ANeuralNetworksCompilation_create(mModel, &compilation), ANEURALNETWORKS_NO_ERROR); in TEST_F()
2588 EXPECT_EQ(ANeuralNetworksBurst_create(compilation, &burst), ANEURALNETWORKS_BAD_STATE); in TEST_F()
2592 ANeuralNetworksCompilation_free(compilation); in TEST_F()
3355 void runExecutionSetTimeoutTest(ANeuralNetworksCompilation* compilation, in runExecutionSetTimeoutTest() argument
3357 if (!compilation) { in runExecutionSetTimeoutTest()
3360 ASSERT_EQ(ANeuralNetworksCompilation_finish(compilation), ANEURALNETWORKS_NO_ERROR); in runExecutionSetTimeoutTest()
3368 ASSERT_EQ(ANeuralNetworksExecution_create(compilation, &execution), in runExecutionSetTimeoutTest()
3418 ASSERT_EQ(ANeuralNetworksBurst_create(compilation, &burst), in runExecutionSetTimeoutTest()
3548 ANeuralNetworksCompilation* compilation; in createAndCompileAddModelWithType() local
3549 EXPECT_EQ(ANeuralNetworksCompilation_create(model, &compilation), ANEURALNETWORKS_NO_ERROR); in createAndCompileAddModelWithType()
3550 EXPECT_EQ(ANeuralNetworksCompilation_finish(compilation), ANEURALNETWORKS_NO_ERROR); in createAndCompileAddModelWithType()
3551 return {model, compilation}; in createAndCompileAddModelWithType()
3554 static void testIncompatibleOperands(const ANeuralNetworksCompilation* compilation, in testIncompatibleOperands() argument
3560 EXPECT_EQ(ANeuralNetworksMemoryDesc_addInputRole(desc, compilation, 0, 1.0f), in testIncompatibleOperands()
3571 EXPECT_EQ(ANeuralNetworksMemoryDesc_addOutputRole(desc, compilation, 0, 1.0f), in testIncompatibleOperands()
3719 ANeuralNetworksCompilation* compilation; in createAndCompileChannelQuantConvModel() local
3720 EXPECT_EQ(ANeuralNetworksCompilation_create(model, &compilation), ANEURALNETWORKS_NO_ERROR); in createAndCompileChannelQuantConvModel()
3721 EXPECT_EQ(ANeuralNetworksCompilation_finish(compilation), ANEURALNETWORKS_NO_ERROR); in createAndCompileChannelQuantConvModel()
3722 return {model, compilation}; in createAndCompileChannelQuantConvModel()