- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for REGISTER_OP (0.09 sec)
-
tensorflow/c/c_api_test.cc
TF_DeleteTensor(fetchValues[1]); } // REGISTER_OP for CApiAttributesTest test cases. // Registers two ops, each with a single attribute called 'v'. // The attribute in one op will have a type 'type', the other // will have list(type). #define ATTR_TEST_REGISTER_OP(type) \ REGISTER_OP("CApiAttributesTestOp" #type) \
Registered: Tue Dec 30 12:39:10 UTC 2025 - Last Modified: Mon Nov 17 00:00:38 UTC 2025 - 97K bytes - Viewed (0) -
tensorflow/BUILD
# visibility = ["//visibility:public"], # ) # copybara:uncomment_end # 'enable_registration_v2' opts-in to a different implementation of op and # kernel registration - REGISTER_OP, REGISTER_KERNEL_BUILDER, etc. # # This setting is currently experimental. The 'v2' implementation does _not_ # correspond to a particular, finalized design; rather, it relates to # developing one. #
Registered: Tue Dec 30 12:39:10 UTC 2025 - Last Modified: Wed Nov 12 19:21:56 UTC 2025 - 53.1K bytes - Viewed (0) -
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,
Registered: Tue Dec 30 12:39:10 UTC 2025 - Last Modified: Mon Nov 17 00:00:38 UTC 2025 - 63.6K bytes - Viewed (1) -
RELEASE.md
* C API gives access to inferred shapes with `TF_GraphGetTensorNumDims` and `TF_GraphGetTensorShape`. * Shape functions for core ops have moved to C++ via `REGISTER_OP(...).SetShapeFn(...)`. Python shape inference RegisterShape calls use the C++ shape functions with `common_shapes.call_cpp_shape_fn`. A future release will remove `RegisterShape` from python.Registered: Tue Dec 30 12:39:10 UTC 2025 - Last Modified: Tue Oct 28 22:27:41 UTC 2025 - 740.4K bytes - Viewed (3)