- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for TestOpWithSingleKernel (0.16 sec)
-
tensorflow/c/c_api_test.cc
void Compute(tensorflow::OpKernelContext* context) override {} }; // Test we can query kernels REGISTER_OP("TestOpWithSingleKernel") .Input("a: float") .Input("b: float") .Output("o: float"); REGISTER_KERNEL_BUILDER( Name("TestOpWithSingleKernel").Device(tensorflow::DEVICE_CPU), DummyKernel); TEST(TestKernel, TestGetAllRegisteredKernels) { TF_Status* status = TF_NewStatus();
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 16:27:48 UTC 2024 - 97K bytes - Viewed (0)