- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for REGISTER_OP (0.1 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) -
tensorflow/c/c_api.h
// The calls to TF_AddInput and TF_AddInputList must match (in number, // order, and type) the op declaration. For example, the "Concat" op // has registration: // REGISTER_OP("Concat") // .Input("concat_dim: int32") // .Input("values: N * T") // .Output("output: T") // .Attr("N: int >= 2") // .Attr("T: type");
Registered: Tue Dec 30 12:39:10 UTC 2025 - Last Modified: Thu Oct 26 21:08:15 UTC 2023 - 82.3K bytes - Viewed (0) -
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)