Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for TFE_OpGetAttrType (0.11 sec)

  1. tensorflow/c/eager/c_api.h

    TF_CAPI_EXPORT extern TF_AttrType TFE_OpGetAttrType(TFE_Op* op,
                                                        const char* attr_name,
                                                        unsigned char* is_list,
                                                        TF_Status* status);
    // Get an attribute type given an op name; a fusion of TFE_NewOp and
    // TFE_OpGetAttrType for use from Python without the overhead of the individual
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Apr 27 21:07:00 UTC 2023
    - 22.8K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api.cc

      TF_AttrType ret;
      TFE_Op* op = TFE_NewOp(ctx, op_or_function_name, status);
      if (status->status.ok()) {
        ret = TFE_OpGetAttrType(op, attr_name, is_list, status);
      } else {
        ret = TF_ATTR_INT;  // Same dummy return as TFE_OpGetAttrType.
      }
      TFE_DeleteOp(op);
      return ret;
    }
    
    void TFE_OpSetAttrString(TFE_Op* op, const char* attr_name, const void* value,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api_test.cc

      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);
      ASSERT_EQ(TF_ATTR_SHAPE,
                TFE_OpGetAttrType(copy_op, "shape", &is_list, status));
      CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Aug 03 20:50:20 UTC 2023
    - 94.6K bytes
    - Viewed (0)
Back to top