- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for TF_GetNodesToPreserveListSize (0.24 sec)
-
tensorflow/c/experimental/grappler/grappler_test.cc
for (const string& s : nodes_preserved) { list_total_size += s.size(); } size_t storage_size = 0; int num_values = 0; TF_Status* status = TF_NewStatus(); TF_GetNodesToPreserveListSize(c_item, &num_values, &storage_size, status); EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); EXPECT_EQ(nodes_preserved.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) -
tensorflow/c/experimental/grappler/grappler.h
// 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); // Get a set of node names that must be preserved. They can not be transformed
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.cc
optimizer_configs, params.device_type); return absl::OkStatus(); } } // namespace grappler } // 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 =
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Sep 06 19:12:29 GMT 2023 - 15K bytes - Viewed (1)