- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for custom (1.45 sec)
-
tensorflow/c/eager/c_api_test.cc
TF_DeleteStatus(status); TFE_DeleteOp(var_op); TFE_DeleteOp(var_op_2); TFE_DeleteContext(ctx); } // Needs to work with a const TFE_Op since custom devices should not modify the // op they are called with. TFE_Op* CloneOp(const TFE_Op* other) { TF_Status* status = TF_NewStatus(); TFE_Context* context = TFE_OpGetContext(other, status);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 20:50:20 GMT 2023 - 94.6K bytes - Viewed (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)