- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for TestOpWithSingleKernel (0.2 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();
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 96.9K bytes - Viewed (3)