Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TF_OperationGetAttrShape (0.16 sec)

  1. tensorflow/c/c_api_test.cc

      TF_SetAttrShape(desc, "v", nullptr, -1);
      auto oper = TF_FinishOperation(desc, s_);
      ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
      EXPECT_TF_META("v", -1, TF_ATTR_SHAPE, -1);
      TF_OperationGetAttrShape(oper, "v", nullptr, 10, s_);
      EXPECT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
    
      // Partially specified shape
      const int64_t partial_shape[] = {17, -1};
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 97K bytes
    - Viewed (0)
  2. tensorflow/c/c_api.h

    // `values` must point to an array of length at least `num_dims` (ideally set to
    // TF_Attr_Meta.size from TF_OperationGetAttrMetadata(oper, attr_name)).
    TF_CAPI_EXPORT extern void TF_OperationGetAttrShape(TF_Operation* oper,
                                                        const char* attr_name,
                                                        int64_t* value,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Oct 26 21:08:15 UTC 2023
    - 82.3K bytes
    - Viewed (0)
Back to top