- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for GrapplerItem (0.2 sec)
-
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) -
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) -
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) -
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)