Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TFE_OpGetOutputLength (0.08 sec)

  1. tensorflow/c/eager/parallel_device/parallel_device.cc

        // 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()) {
          std::string message(absl::StrCat(
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 18.3K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api.cc

                                                   TF_Status* status) {
      int ret = -1;
      status->status = tensorflow::unwrap(op)->InputLength(input_name, &ret);
      return ret;
    }
    
    TF_CAPI_EXPORT extern int TFE_OpGetOutputLength(TFE_Op* op,
                                                    const char* output_name,
                                                    TF_Status* status) {
      int ret = -1;
    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