Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TF_SetAttrShapeList (0.47 sec)

  1. tensorflow/c/c_api.h

    // Set `num_dims[i]` to -1 to represent "unknown rank".  Otherwise,
    // `dims[i]` points to an array of length `num_dims[i]`.  `dims[i][j]`
    // must be >= -1, with -1 meaning "unknown dimension".
    TF_CAPI_EXPORT extern void TF_SetAttrShapeList(TF_OperationDescription* desc,
                                                   const char* attr_name,
                                                   const int64_t* const* dims,
    Registered: Tue Sep 09 12:39:10 UTC 2025
    - Last Modified: Thu Oct 26 21:08:15 UTC 2023
    - 82.3K bytes
    - Viewed (0)
  2. tensorflow/c/c_api_test.cc

      const size_t list_size = TF_ARRAYSIZE(list);
      const int ndims[] = {TF_ARRAYSIZE(shape_1), TF_ARRAYSIZE(shape_2)};
      const int total_ndims = 5;  // ndims[0] + ndims[1]
    
      auto desc = init("list(shape)");
      TF_SetAttrShapeList(desc, "v", list, ndims, list_size);
      auto oper = TF_FinishOperation(desc, s_);
      ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
    
      EXPECT_TF_META("v", list_size, TF_ATTR_SHAPE, total_ndims);
    Registered: Tue Sep 09 12:39:10 UTC 2025
    - Last Modified: Fri Dec 27 12:18:10 UTC 2024
    - 97K bytes
    - Viewed (0)
Back to top