Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for TF_GetNodesToPreserveList (0.27 sec)

  1. tensorflow/c/experimental/grappler/grappler.h

    // or removed during the graph transformation. This includes feed and fetch
    // nodes, keep_ops, init_ops. Fills in `num_values` and `storage_size`, they
    // will be used in `TF_GetNodesToPreserveList`.
    TF_CAPI_EXPORT extern void TF_GetNodesToPreserveListSize(
        const TF_GrapplerItem* item, int* num_values, size_t* storage_size,
        TF_Status* status);
    
    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)
  2. tensorflow/c/experimental/grappler/grappler_test.cc

      std::unique_ptr<char*[]> values(new char*[nodes_preserved.size()]);
      std::unique_ptr<size_t[]> lens(new size_t[nodes_preserved.size()]);
      std::unique_ptr<char[]> storage(new char[storage_size]);
      TF_GetNodesToPreserveList(c_item, values.get(), lens.get(),
                                nodes_preserved.size(), storage.get(), storage_size,
                                status);
      EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    
    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)
  3. tensorflow/c/experimental/grappler/grappler.cc

              ->NodesToPreserve();
      *num_values = nodes.size();
      *storage_size = 0;
      for (const std::string& str : nodes) {
        *storage_size += str.size();
      }
    }
    
    void TF_GetNodesToPreserveList(const TF_GrapplerItem* item, char** values,
                                   size_t* lengths, int num_values, void* storage,
                                   size_t storage_size, TF_Status* status) {
    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