- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for REGISTER_OP (0.23 sec)
-
tensorflow/c/experimental/ops/README.md
category names correspond to generated source file names, and should be consistent with the original source files registering each operator. For example since `REGISTER_OP("MatMul")` appears in ***core/math_ops.cc***, the "MatMul" operator in the script should be in the "math" category, and it will be generated in the output file `c/experimental/ops/math_ops.cc`.
Plain Text - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Jul 28 17:21:01 GMT 2021 - 993 bytes - Viewed (0) -
tensorflow/c/c_api_test.cc
REGISTER_OP("CApiAttributesTestOpList" #type) \ .Attr("v: list(" #type ")") \ .SetShapeFn(tensorflow::shape_inference::UnknownShape) ATTR_TEST_REGISTER_OP(string); ATTR_TEST_REGISTER_OP(int); ATTR_TEST_REGISTER_OP(float); ATTR_TEST_REGISTER_OP(bool); ATTR_TEST_REGISTER_OP(type); ATTR_TEST_REGISTER_OP(shape);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 96.9K bytes - Viewed (3) -
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. #
Plain Text - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Apr 09 18:15:11 GMT 2024 - 53.4K bytes - Viewed (8) -
tensorflow/c/eager/c_api_test.cc
} TEST(CAPI, ExecuteWithTracing) { ExecuteWithTracing(false); } TEST(CAPI, ExecuteWithTracingAsync) { ExecuteWithTracing(true); } REGISTER_OP("TestNonCommUnavailable") .Output("out: string") .Doc(R"doc(Test non-communication op throwing Unavailable error.)doc"); REGISTER_OP("TestCommUnavailable") .Output("out: string") .SetIsDistributedCommunication()
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) -
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");
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Oct 26 21:08:15 GMT 2023 - 82.3K bytes - Viewed (3) -
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.
Plain Text - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 29 19:17:57 GMT 2024 - 727.7K bytes - Viewed (8)