Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for TestOpWithSingleKernel (0.16 sec)

  1. 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)
Back to top