Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for TF_OperationGetAttrType (0.22 sec)

  1. tensorflow/c/c_api.h

                                                           int max_values,
                                                           TF_Status* status);
    
    TF_CAPI_EXPORT extern void TF_OperationGetAttrType(TF_Operation* oper,
                                                       const char* attr_name,
                                                       TF_DataType* value,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Oct 26 21:08:15 UTC 2023
    - 82.3K bytes
    - Viewed (0)
  2. tensorflow/c/c_api.cc

      }
    DEFINE_GETATTR(TF_OperationGetAttrInt, int64_t, int64_t, i);
    DEFINE_GETATTR(TF_OperationGetAttrFloat, float, float, f);
    DEFINE_GETATTR(TF_OperationGetAttrBool, unsigned char, bool, b);
    DEFINE_GETATTR(TF_OperationGetAttrType, TF_DataType, DataType, type);
    #undef DEFINE_GETATTR
    
    void TF_OperationGetAttrShape(TF_Operation* oper, const char* attr_name,
                                  int64_t* value, int num_dims, TF_Status* status) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  3. tensorflow/c/c_api_test.cc

      auto oper = TF_FinishOperation(desc, s_);
      ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
      EXPECT_TF_META("v", -1, TF_ATTR_TYPE, -1);
    
      TF_DataType value;
      TF_OperationGetAttrType(oper, "v", &value, s_);
      EXPECT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
      EXPECT_EQ(TF_COMPLEX128, value);
    }
    
    TEST_F(CApiAttributesTest, TypeList) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 96.9K bytes
    - Viewed (0)
Back to top