Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for custom (0.17 sec)

  1. tensorflow/c/c_api_function_test.cc

                                 /*opts=*/nullptr, description, s.get());
      ASSERT_EQ(TF_OK, TF_GetCode(s.get())) << TF_Message(s.get());
      ASSERT_NE(*func, nullptr);
    }
    
    REGISTER_OP("CustomOp")
        .Output("output: float32")
        .Attr("index: int")
        .SetShapeFn(tensorflow::shape_inference::UnknownShape);
    
    void NodeWithPlaceholderAttrHelper(TF_Graph* graph, TF_Status* s,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jul 20 22:08:54 GMT 2023
    - 63.6K bytes
    - Viewed (6)
Back to top