Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for TFE_OpAddAttrs (0.22 sec)

  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;
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Wed Mar 03 20:47:31 GMT 2021
    - 8.3K 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()) {
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Wed Mar 29 22:05:31 GMT 2023
    - 18.3K bytes
    - Viewed (0)
  3. 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;
      }
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Fri Feb 09 07:47:20 GMT 2024
    - 25.4K bytes
    - Viewed (1)
  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,
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 43.9K bytes
    - Viewed (2)
  5. 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.
    C
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Wed Feb 21 22:37:46 GMT 2024
    - 39.5K bytes
    - Viewed (0)
  6. 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);
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Aug 03 20:50:20 GMT 2023
    - 94.6K bytes
    - Viewed (1)
Back to top