Searched refs:OpKernelBuilderTest (Results 1 – 1 of 1) sorted by relevance
404 class OpKernelBuilderTest : public ::testing::Test { class521 TEST_F(OpKernelBuilderTest, BuilderCPU) { in TEST_F() argument531 TEST_F(OpKernelBuilderTest, BuilderGPU) { in TEST_F() argument540 TEST_F(OpKernelBuilderTest, BuilderBoth) { in TEST_F() argument550 TEST_F(OpKernelBuilderTest, BuilderTypeAttr) { in TEST_F() argument564 TEST_F(OpKernelBuilderTest, BuilderTypeListAttr) { in TEST_F() argument596 TEST_F(OpKernelBuilderTest, DuplicateKernel) { in TEST_F() argument615 TEST_F(OpKernelBuilderTest, DuplicateKernelForT) { in TEST_F() argument637 TEST_F(OpKernelBuilderTest, BadConstraint) { in TEST_F() argument655 TEST_F(OpKernelBuilderTest, OpOutputList) { in TEST_F() argument[all …]