Searched refs:CountOpsExecutedByCpuKernel (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_device_selection_test.cc | 194 EXPECT_EQ(m.CountOpsExecutedByCpuKernel(), 1); in TEST_F()
208 EXPECT_EQ(m.CountOpsExecutedByCpuKernel(), 1); in TEST_F()
301 EXPECT_EQ(m.CountOpsExecutedByCpuKernel(), 1) in TEST_F()
312 EXPECT_EQ(m1.CountOpsExecutedByCpuKernel(), 0) in TEST_F()
337 EXPECT_EQ(m.CountOpsExecutedByCpuKernel(), 1) in TEST_F()
347 EXPECT_EQ(m1.CountOpsExecutedByCpuKernel(), 0) in TEST_F()
359 EXPECT_EQ(m2.CountOpsExecutedByCpuKernel(), 0) in TEST_F()
465 EXPECT_EQ(m.CountOpsExecutedByCpuKernel(), 1); in TEST_F()
495 EXPECT_EQ(m.CountOpsExecutedByCpuKernel(), 2); in TEST_F()
528 ASSERT_EQ(m.CountOpsExecutedByCpuKernel(), 0); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | test_util.cc | 377 int SingleOpModel::CountOpsExecutedByCpuKernel() { in CountOpsExecutedByCpuKernel() function in tflite::SingleOpModel
|
| D | test_util.h | 616 int CountOpsExecutedByCpuKernel();
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | lstm_full_test.cc | 172 ~LSTMOpModel() { EXPECT_EQ(CountOpsExecutedByCpuKernel(), 0); } in ~LSTMOpModel()
|