Search Options

Display Count
Sort
Preferred Language
Advanced Search

Results 1 - 4 of 4 for TFE_OpAddAttrs (0.06 seconds)

  1. 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)
  2. 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)
  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,
    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)
  4. 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)
Back to Top