Searched refs:GetKernelClassName (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel_test.cc | 495 string GetKernelClassName(const string& op_type, in GetKernelClassName() function in tensorflow::__anon843475bc0111::OpKernelBuilderTest
523 EXPECT_EQ("DummyKernel", GetKernelClassName("BuildCPU", DEVICE_CPU, {})); in TEST_F()
525 EXPECT_EQ("not found", GetKernelClassName("BuildCPU", DEVICE_GPU, {})); in TEST_F()
566 EXPECT_EQ("DummyKernel", GetKernelClassName("BuildTypeListAttr", DEVICE_CPU, in TEST_F()
570 EXPECT_EQ("DummyKernel", GetKernelClassName("BuildTypeListAttr", DEVICE_CPU, in TEST_F()
578 EXPECT_EQ("not found", GetKernelClassName("BuildTypeListAttr", DEVICE_CPU, in TEST_F()
583 GetKernelClassName("BuildTypeListAttr", DEVICE_CPU, {}), in TEST_F()
915 GetKernelClassName("LabeledKernel", DEVICE_CPU, {})); in TEST_F()
923 EXPECT_EQ("LabeledKernel<1>", GetKernelClassName("LabeledKernel", DEVICE_CPU, in TEST_F()
|