- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for TFE_OpAddAttrs (0.16 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
if (TF_GetCode(status) != TF_OK) return; } else { TFE_OpReset(op_.get(), operation_name, device_.c_str(), status); if (TF_GetCode(status) != TF_OK) return; } TFE_OpAddAttrs(op_.get(), attributes); for (int input_index = 0; input_index < inputs.size(); ++input_index) { TFE_OpAddInput(op_.get(), inputs[input_index], status); if (TF_GetCode(status) != TF_OK) return; }
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Mon Oct 21 04:14:14 UTC 2024 - 25.9K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device.cc
// Special-cased operation for un-packing one parallel tensor into // per-device tensors. OpPtr op(TFE_NewOp(context, operation_name, status)); TFE_OpAddAttrs(op.get(), attributes); int expected_outputs = TFE_OpGetOutputLength(op.get(), "outputs", status); if (TF_GetCode(status) != TF_OK) return result; if (expected_outputs != parallel_device.num_underlying_devices()) {
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Mon Oct 21 04:14:14 UTC 2024 - 18.3K bytes - Viewed (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.
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Wed Feb 21 22:37:46 UTC 2024 - 39.5K bytes - Viewed (0) -
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,
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 43.9K bytes - Viewed (0)