Searched refs:kernel_list (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/python/framework/ |
D | kernels.py | 31 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
|
D | kernels_test.py | 28 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/ |
D | op_kernel_test.cc | 1001 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()
|
D | op_kernel.cc | 1208 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/ |
D | c_api_test.cc | 2467 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 …]
|
D | c_api.cc | 2878 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()
|