- Sort Score
- Num 10 results
- Language All
Results 1 - 4 of 4 for TFE_OpAddAttrs (0.06 seconds)
-
tensorflow/c/eager/custom_device_testutil.cc
if (TF_GetCode(s) != TF_OK) return; const TFE_OpAttrs* attributes = TFE_OpGetAttrs(original_op); TFE_Op* op(TFE_NewOp(context, operation_name, s)); if (TF_GetCode(s) != TF_OK) return; TFE_OpAddAttrs(op, attributes); TFE_OpSetDevice(op, dev->underlying_device.c_str(), s); if (TF_GetCode(s) != TF_OK) return; int num_inputs = TFE_OpGetFlatInputCount(original_op, s); if (TF_GetCode(s) != TF_OK) return;
Created: Tue Apr 07 12:39:13 GMT 2026 - Last Modified: Sat Dec 20 23:30:21 GMT 2025 - 8.4K bytes - Click Count (0) -
tensorflow/c/eager/c_api_experimental.h
TF_CAPI_EXPORT extern const TFE_OpAttrs* TFE_OpGetAttrs(const TFE_Op* op); // Add attributes in `attrs` to `op`. // // Does not overwrite or update existing attributes, but adds new ones. TF_CAPI_EXPORT extern void TFE_OpAddAttrs(TFE_Op* op, const TFE_OpAttrs* attrs); // Serialize `attrs` as a tensorflow::NameAttrList protocol buffer (into `buf`), // containing the op name and a map of its attributes.
Created: Tue Apr 07 12:39:13 GMT 2026 - Last Modified: Wed Feb 21 22:37:46 GMT 2024 - 39.5K bytes - Click Count (1) -
tensorflow/c/eager/c_api.cc
tensorflow::unwrap(ctx)->EndStep(); } const TFE_OpAttrs* TFE_OpGetAttrs(const TFE_Op* op) { return tensorflow::wrap(tensorflow::unwrap(op)->GetOpAttrs()); } void TFE_OpAddAttrs(TFE_Op* op, const TFE_OpAttrs* attrs) { tensorflow::unwrap(op)->AddAttrs(tensorflow::unwrap(attrs)); } void TFE_OpAttrsSerialize(const TFE_OpAttrs* attrs, TF_Buffer* buf,
Created: Tue Apr 07 12:39:13 GMT 2026 - Last Modified: Fri Nov 07 05:55:21 GMT 2025 - 43.9K bytes - Click Count (0) -
tensorflow/c/eager/c_api_test.cc
const TFE_OpAttrs* attributes = TFE_OpGetAttrs(var_op); TFE_Op* copy_op = TFE_NewOp(ctx, "VarHandleOp", status); TFE_OpSetAttrType(copy_op, "dtype", TF_FLOAT); TFE_OpAddAttrs(copy_op, attributes); unsigned char is_list = 0; ASSERT_EQ(TF_ATTR_TYPE, TFE_OpGetAttrType(copy_op, "dtype", &is_list, status)); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
Created: Tue Apr 07 12:39:13 GMT 2026 - Last Modified: Thu Oct 09 05:56:18 GMT 2025 - 94.6K bytes - Click Count (0)