Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for TFE_OpAddAttrs (0.16 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
Back to top