| 1130 | BENCHMARK(BM_SelectInputRange); |
| 1131 | |
| 1132 | TEST(RegisteredKernels, CanCallGetAllRegisteredKernels) { |
| 1133 | auto kernel_list = GetAllRegisteredKernels(); |
| 1134 | auto all_registered_kernels = kernel_list.kernel(); |
| 1135 | auto has_name_test1 = [](const KernelDef& k) { return k.op() == "Test1"; }; |
| 1136 | |
| 1137 | // Verify we can find the "Test1" op registered above |
| 1138 | auto test1_it = std::find_if(all_registered_kernels.begin(), |
| 1139 | all_registered_kernels.end(), has_name_test1); |
| 1140 | ASSERT_NE(test1_it, all_registered_kernels.end()); |
| 1141 | EXPECT_EQ(test1_it->device_type(), "CPU"); |
| 1142 | |
| 1143 | // Verify there was just one kernel |
| 1144 | ++test1_it; |
| 1145 | EXPECT_EQ( |
| 1146 | std::find_if(test1_it, all_registered_kernels.end(), has_name_test1), |
| 1147 | all_registered_kernels.end()); |
| 1148 | } |
| 1149 | |
| 1150 | // Simple test just to check we can call LogAllRegisteredKernels |
| 1151 | TEST(RegisteredKernels, CanLogAllRegisteredKernels) { |
nothing calls this directly
no test coverage detected