Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for GrapplerItem (0.2 sec)

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

                "'optimize_func' field in TP_Optimizer must be set.");
    }
    
    TEST(TF_GrapplerItem, NodesToPreserve) {
      GrapplerItem item;
      item.fetch = std::vector<string>{"Conv", "BiasAdd"};
      std::unordered_set<string> nodes_preserved = item.NodesToPreserve();
      TF_GrapplerItem* c_item = reinterpret_cast<TF_GrapplerItem*>(&item);
    
      int list_total_size = 0;
      for (const string& s : nodes_preserved) {
    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.cc

    }  // namespace tensorflow
    
    void TF_GetNodesToPreserveListSize(const TF_GrapplerItem* item, int* num_values,
                                       size_t* storage_size, TF_Status* status) {
      TF_SetStatus(status, TF_OK, "");
      const std::unordered_set<std::string>& nodes =
          reinterpret_cast<const tensorflow::grappler::GrapplerItem*>(item)
              ->NodesToPreserve();
      *num_values = nodes.size();
      *storage_size = 0;
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Wed Sep 06 19:12:29 GMT 2023
    - 15K bytes
    - Viewed (1)
  3. tensorflow/c/experimental/grappler/grappler.h

    typedef enum TF_TriState {
      TF_TriState_Default = 0,
      TF_TriState_Off,
      TF_TriState_On,
    } TF_TriState;
    
    // TF_GrapplerItem represents a combination of a graph, one of more fetch nodes,
    // and potentially a set of nodes to feed.
    typedef struct TF_GrapplerItem TF_GrapplerItem;
    
    // Flags indicating whether existing optimizers should be turned off.
    // It's optional for plugin to set functions to return true/false. If not
    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)
  4. tensorflow/c/experimental/grappler/grappler_internal.h

    // Registers Graph optimizers.
    Status InitGraphPlugin(void* dso_handle);
    
    // Allow registering a graph optimizer using a function (used for
    // testing).
    Status InitGraphPlugin(TFInitGraphPluginFn init_fn);
    
    struct GrapplerItem;
    class Cluster;
    
    struct TFStatusDeleter {
      void operator()(TF_Status* s) const { TF_DeleteStatus(s); }
    };
    using OwnedTFStatus = std::unique_ptr<TF_Status, TFStatusDeleter>;
    
    struct TFBufferDeleter {
    C
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Wed Jun 08 08:58:23 GMT 2022
    - 3.5K bytes
    - Viewed (1)
Back to top