Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for TestOpWithSingleKernel (0.2 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();
    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)
Back to top