Search Options

Results per page
Sort
Preferred Languages
Advance

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

  1. tensorflow/c/c_api.h

    // TF_AttrMetadata.total_size from TF_OperationGetAttrMetadata(oper,
    // attr_name).
    //
    // Fails if storage_size is insufficient to hold the requested shapes.
    TF_CAPI_EXPORT extern void TF_OperationGetAttrShapeList(
        TF_Operation* oper, const char* attr_name, int64_t** dims, int* num_dims,
        int num_shapes, int64_t* storage, int storage_size, TF_Status* status);
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Oct 26 21:08:15 UTC 2023
    - 82.3K bytes
    - Viewed (0)
  2. tensorflow/c/c_api.cc

      if (!status->status.ok()) return;
      auto len = std::min(shape.dims(), num_dims);
      for (int i = 0; i < len; ++i) {
        value[i] = shape.dim_size(i);
      }
    }
    
    void TF_OperationGetAttrShapeList(TF_Operation* oper, const char* attr_name,
                                      int64_t** dims, int* num_dims, int num_shapes,
                                      int64_t* storage, int storage_size,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  3. tensorflow/c/c_api_test.cc

      EXPECT_TF_META("v", list_size, TF_ATTR_SHAPE, total_ndims);
      int64_t* values[list_size];
      int values_ndims[list_size];
      int64_t storage[total_ndims];
      TF_OperationGetAttrShapeList(oper, "v", values, values_ndims, list_size,
                                   storage, total_ndims, s_);
      EXPECT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
      for (size_t i = 0; i < list_size; ++i) {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 97K bytes
    - Viewed (0)
Back to top