Home
last modified time | relevance | path

Searched refs:kernel_list (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/framework/
Dkernels.py31 kernel_list = kernel_def_pb2.KernelList()
32 kernel_list.ParseFromString(compat.as_bytes(data))
33 return kernel_list
44 kernel_list = kernel_def_pb2.KernelList()
45 kernel_list.ParseFromString(compat.as_bytes(data))
46 return kernel_list
Dkernels_test.py28 kernel_list = kernels.get_all_registered_kernels()
29 self.assertGreater(len(kernel_list.kernel), 0)
35 kernel_list = kernels.get_registered_kernels_for_op("KernelLabel")
36 self.assertGreater(len(kernel_list.kernel), 0)
37 self.assertEqual(kernel_list.kernel[0].op, "KernelLabel")
/external/tensorflow/tensorflow/core/framework/
Dop_kernel_test.cc1001 auto kernel_list = GetAllRegisteredKernels(); in TEST() local
1002 auto all_registered_kernels = kernel_list.kernel(); in TEST()
1025 auto kernel_list = GetFilteredRegisteredKernels(has_name_test1); in TEST() local
1026 ASSERT_EQ(kernel_list.kernel_size(), 1); in TEST()
1027 EXPECT_EQ(kernel_list.kernel(0).op(), "Test1"); in TEST()
1028 EXPECT_EQ(kernel_list.kernel(0).device_type(), "CPU"); in TEST()
1032 auto kernel_list = GetRegisteredKernelsForOp("Test1"); in TEST() local
1033 ASSERT_EQ(kernel_list.kernel_size(), 1); in TEST()
1034 EXPECT_EQ(kernel_list.kernel(0).op(), "Test1"); in TEST()
1035 EXPECT_EQ(kernel_list.kernel(0).device_type(), "CPU"); in TEST()
Dop_kernel.cc1208 KernelList kernel_list = GetAllRegisteredKernels(); in LogAllRegisteredKernels() local
1209 for (const auto& kernel_def : kernel_list.kernel()) { in LogAllRegisteredKernels()
1221 KernelList kernel_list; in GetFilteredRegisteredKernels() local
1222 kernel_list.mutable_kernel()->Reserve(typed_registry->size()); in GetFilteredRegisteredKernels()
1226 *kernel_list.add_kernel() = kernel_def; in GetFilteredRegisteredKernels()
1229 return kernel_list; in GetFilteredRegisteredKernels()
1238 KernelList kernel_list = GetRegisteredKernelsForOp(op_name); in KernelsRegisteredForOp() local
1239 if (kernel_list.kernel_size() == 0) return " <no registered kernels>\n"; in KernelsRegisteredForOp()
1241 for (const auto& kernel_def : kernel_list.kernel()) { in KernelsRegisteredForOp()
/external/tensorflow/tensorflow/c/
Dc_api_test.cc2467 KernelList kernel_list; in TEST() local
2468 kernel_list.ParseFromArray(kernel_list_buf->data, kernel_list_buf->length); in TEST()
2469 ASSERT_GT(kernel_list.kernel_size(), 0); in TEST()
2479 KernelList kernel_list; in TEST() local
2480 kernel_list.ParseFromArray(kernel_list_buf->data, kernel_list_buf->length); in TEST()
2481 ASSERT_EQ(kernel_list.kernel_size(), 1); in TEST()
2482 EXPECT_EQ(kernel_list.kernel(0).op(), "TestOpWithSingleKernel"); in TEST()
2483 EXPECT_EQ(kernel_list.kernel(0).device_type(), "CPU"); in TEST()
2492 KernelList kernel_list; in TEST() local
2493 kernel_list.ParseFromArray(kernel_list_buf->data, kernel_list_buf->length); in TEST()
[all …]
Dc_api.cc2878 tensorflow::KernelList kernel_list = tensorflow::GetAllRegisteredKernels(); in TF_GetAllRegisteredKernels() local
2880 status->status = MessageToBuffer(kernel_list, ret); in TF_GetAllRegisteredKernels()
2889 tensorflow::KernelList kernel_list = in TF_GetRegisteredKernelsForOp() local
2892 status->status = MessageToBuffer(kernel_list, ret); in TF_GetRegisteredKernelsForOp()