Searched refs:CountOpsExecutedByCpuKernel (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
D | nnapi_delegate_device_selection_test.cc | 211 EXPECT_EQ(m.CountOpsExecutedByCpuKernel(), 1); in TEST_F() 234 EXPECT_EQ(m.CountOpsExecutedByCpuKernel(), 1); in TEST_F() 311 EXPECT_EQ(m.CountOpsExecutedByCpuKernel(), 1) in TEST_F() 322 EXPECT_EQ(m1.CountOpsExecutedByCpuKernel(), 0) in TEST_F() 336 EXPECT_EQ(m.CountOpsExecutedByCpuKernel(), 1) in TEST_F() 345 EXPECT_EQ(m1.CountOpsExecutedByCpuKernel(), 0) in TEST_F() 356 EXPECT_EQ(m2.CountOpsExecutedByCpuKernel(), 0) in TEST_F() 461 EXPECT_EQ(m.CountOpsExecutedByCpuKernel(), 1); in TEST_F() 490 EXPECT_EQ(m.CountOpsExecutedByCpuKernel(), 2); in TEST_F() 522 ASSERT_EQ(m.CountOpsExecutedByCpuKernel(), 0); in TEST_F()
|
/external/tensorflow/tensorflow/lite/kernels/ |
D | test_util.cc | 341 int SingleOpModel::CountOpsExecutedByCpuKernel() { in CountOpsExecutedByCpuKernel() function in tflite::SingleOpModel
|
D | test_util.h | 417 int CountOpsExecutedByCpuKernel();
|