Home
last modified time | relevance | path

Searched refs:kernel_list (Results 1 – 7 of 7) 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/lite/delegates/flex/
Dallowlisted_flex_ops_test.cc32 tensorflow::KernelList kernel_list = in GetAllCpuKernels() local
36 for (int i = 0; i < kernel_list.kernel_size(); ++i) { in GetAllCpuKernels()
37 tensorflow::KernelDef kernel_def = kernel_list.kernel(i); in GetAllCpuKernels()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel_test.cc1110 auto kernel_list = GetAllRegisteredKernels(); in TEST() local
1111 auto all_registered_kernels = kernel_list.kernel(); in TEST()
1134 auto kernel_list = GetFilteredRegisteredKernels(has_name_test1); in TEST() local
1135 ASSERT_EQ(kernel_list.kernel_size(), 1); in TEST()
1136 EXPECT_EQ(kernel_list.kernel(0).op(), "Test1"); in TEST()
1137 EXPECT_EQ(kernel_list.kernel(0).device_type(), "CPU"); in TEST()
1141 auto kernel_list = GetRegisteredKernelsForOp("Test1"); in TEST() local
1142 ASSERT_EQ(kernel_list.kernel_size(), 1); in TEST()
1143 EXPECT_EQ(kernel_list.kernel(0).op(), "Test1"); in TEST()
1144 EXPECT_EQ(kernel_list.kernel(0).device_type(), "CPU"); in TEST()
Dop_kernel.cc1525 KernelList kernel_list = GetAllRegisteredKernels(); in LogAllRegisteredKernels() local
1526 for (const auto& kernel_def : kernel_list.kernel()) { in LogAllRegisteredKernels()
1538 KernelList kernel_list; in GetFilteredRegisteredKernels() local
1540 kernel_list.mutable_kernel()->Reserve(typed_registry->registry.size()); in GetFilteredRegisteredKernels()
1544 *kernel_list.add_kernel() = kernel_def; in GetFilteredRegisteredKernels()
1547 return kernel_list; in GetFilteredRegisteredKernels()
1556 KernelList kernel_list = GetRegisteredKernelsForOp(op_name); in KernelsRegisteredForOp() local
1557 if (kernel_list.kernel_size() == 0) return " <no registered kernels>\n"; in KernelsRegisteredForOp()
1559 for (const auto& kernel_def : kernel_list.kernel()) { in KernelsRegisteredForOp()
/external/tensorflow/tensorflow/c/
Dc_api_test.cc2514 KernelList kernel_list; in TEST() local
2515 kernel_list.ParseFromArray(kernel_list_buf->data, kernel_list_buf->length); in TEST()
2516 ASSERT_GT(kernel_list.kernel_size(), 0); in TEST()
2526 KernelList kernel_list; in TEST() local
2527 kernel_list.ParseFromArray(kernel_list_buf->data, kernel_list_buf->length); in TEST()
2528 ASSERT_EQ(kernel_list.kernel_size(), 1); in TEST()
2529 EXPECT_EQ(kernel_list.kernel(0).op(), "TestOpWithSingleKernel"); in TEST()
2530 EXPECT_EQ(kernel_list.kernel(0).device_type(), "CPU"); in TEST()
2539 KernelList kernel_list; in TEST() local
2540 kernel_list.ParseFromArray(kernel_list_buf->data, kernel_list_buf->length); in TEST()
[all …]
Dc_api.cc2470 tensorflow::KernelList kernel_list = tensorflow::GetAllRegisteredKernels(); in TF_GetAllRegisteredKernels() local
2472 status->status = MessageToBuffer(kernel_list, ret); in TF_GetAllRegisteredKernels()
2481 tensorflow::KernelList kernel_list = in TF_GetRegisteredKernelsForOp() local
2484 status->status = MessageToBuffer(kernel_list, ret); in TF_GetRegisteredKernelsForOp()