Searched defs:TestKernel (Results 1 – 4 of 4) sorted by relevance
152 TEST(TestKernel, TestRegisterKernelBuilder) { in TEST() argument192 TEST(TestKernel, TF_RegisterKernelBuilderWithKernelDef) { in TEST() argument745 TEST(TestKernel, TestInputAndOutputCount) { in TEST() argument837 TEST(TestKernel, DeleteKernelBuilderIsOkOnNull) { in TEST() argument912 TEST(TestKernel, TestHostMemory) { in TEST() argument
2559 TEST(TestKernel, TestGetAllRegisteredKernels) { in TEST() argument2570 TEST(TestKernel, TestGetRegisteredKernelsForOp) { in TEST() argument2584 TEST(TestKernel, TestGetRegisteredKernelsForOpNoKernels) { in TEST() argument
129 class TestKernel : public TFRTOpKernel { class131 explicit TestKernel(TFRTOpKernelConstruction* construction) in TestKernel() function in tensorflow::__anon10b29e460111::TestKernel
32 class TestKernel : public OpKernel { class34 explicit TestKernel(OpKernelConstruction* context) : OpKernel(context) {} in TestKernel() function in tensorflow::__anon0c04c7cb0111::TestKernel