Search Options

Results per page
Sort
Preferred Languages
Advance

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

  1. tensorflow/c/experimental/grappler/grappler_test.cc

      int list_total_size = 0;
      for (const string& s : item.fetch) {
        list_total_size += s.size();
      }
    
      size_t storage_size = 0;
      int num_values = 0;
      TF_Status* status = TF_NewStatus();
      TF_GetFetchNodesListSize(c_item, &num_values, &storage_size, status);
      EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      EXPECT_EQ(item.fetch.size(), num_values);
      EXPECT_EQ(list_total_size, storage_size);
    
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Thu Apr 13 22:30:58 GMT 2023
    - 11.6K bytes
    - Viewed (0)
  2. tensorflow/c/experimental/grappler/grappler.h

        void* storage, size_t storage_size, TF_Status* status);
    
    // Get a set of node names for fetch nodes. Fills in `values` and `lengths`,
    // they will be used in `TF_GetFetchNodesList`
    TF_CAPI_EXPORT extern void TF_GetFetchNodesListSize(const TF_GrapplerItem* item,
                                                        int* num_values,
                                                        size_t* storage_size,
    C
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Wed Aug 03 18:08:43 GMT 2022
    - 12.5K bytes
    - Viewed (0)
  3. tensorflow/c/experimental/grappler/grappler.cc

                  "Not enough storage to hold the requested list of nodes"));
          return;
        }
        memcpy(values[index], s.data(), s.size());
        p += s.size();
        index++;
      }
    }
    
    void TF_GetFetchNodesListSize(const TF_GrapplerItem* item, int* num_values,
                                  size_t* storage_size, TF_Status* status) {
      TF_SetStatus(status, TF_OK, "");
      const std::vector<std::string>& nodes =
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Wed Sep 06 19:12:29 GMT 2023
    - 15K bytes
    - Viewed (1)
Back to top