Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for TFE_OpAddAttrs (0.14 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.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