Searched refs:CountOpsExecutedByCpuKernel (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_device_selection_test.cc | 196 EXPECT_EQ(m.CountOpsExecutedByCpuKernel(), 1); in TEST_F()
210 EXPECT_EQ(m.CountOpsExecutedByCpuKernel(), 1); in TEST_F()
302 EXPECT_EQ(m.CountOpsExecutedByCpuKernel(), 1) in TEST_F()
313 EXPECT_EQ(m1.CountOpsExecutedByCpuKernel(), 0) in TEST_F()
338 EXPECT_EQ(m.CountOpsExecutedByCpuKernel(), 1) in TEST_F()
348 EXPECT_EQ(m1.CountOpsExecutedByCpuKernel(), 0) in TEST_F()
360 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/delegates/gpu/cl/kernels/ |
| D | lstm_full_test.cc | 172 ~LSTMOpModel() { EXPECT_EQ(CountOpsExecutedByCpuKernel(), 0); } in ~LSTMOpModel()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | test_util.cc | 412 int SingleOpModel::CountOpsExecutedByCpuKernel() { in CountOpsExecutedByCpuKernel() function in tflite::SingleOpModel
|
| D | test_util.h | 668 int CountOpsExecutedByCpuKernel();
|