MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / TEST

Function TEST

tensorflow/core/framework/op_kernel_test.cc:1132–1148  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1130BENCHMARK(BM_SelectInputRange);
1131
1132TEST(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
1151TEST(RegisteredKernels, CanLogAllRegisteredKernels) {

Callers

nothing calls this directly

Calls 10

GetAllRegisteredKernelsFunction · 0.85
LogAllRegisteredKernelsFunction · 0.85
kernel_sizeMethod · 0.80
kernelMethod · 0.45
opMethod · 0.45
beginMethod · 0.45
endMethod · 0.45
device_typeMethod · 0.45

Tested by

no test coverage detected