- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for TP_Optimizer (0.15 sec)
-
tensorflow/c/experimental/grappler/grappler.h
// [Optional] // Destroy function for optimizer. If Create function is provided, destroy // function is must. void (*destroy_func)(void*); } TP_Optimizer; #define TP_OPTIMIZER_STRUCT_SIZE TF_OFFSET_OF_END(TP_Optimizer, destroy_func) typedef struct TP_OptimizerRegistrationParams { size_t struct_size; void* ext; // reserved for future use // Graph C API version.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 03 18:08:43 UTC 2022 - 12.5K bytes - Viewed (0) -
tensorflow/c/experimental/grappler/grappler_internal.h
}; using OwnedTFBuffer = std::unique_ptr<TF_Buffer, TFBufferDeleter>; class CGraphOptimizer : public CustomGraphOptimizer { public: explicit CGraphOptimizer(TP_Optimizer optimizer, const char* device_type) : optimizer_(optimizer), device_type_(device_type) { if (optimizer.create_func != nullptr) { c_optimizer_ = (*optimizer_.create_func)(); } else {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 08 08:58:23 UTC 2022 - 3.5K bytes - Viewed (0) -
tensorflow/c/experimental/grappler/grappler.cc
VALIDATE_MEMBER(TP_OptimizerRegistrationParams, params, device_type); return absl::OkStatus(); } absl::Status ValidateTPOptimizer(const TP_Optimizer& optimizer) { VALIDATE_STRUCT_SIZE(TP_Optimizer, optimizer, TP_OPTIMIZER_STRUCT_SIZE); VALIDATE_MEMBER(TP_Optimizer, optimizer, optimize_func); return absl::OkStatus(); } absl::Status ValidateTPOptimizerConfigs(const TP_OptimizerConfigs& configs) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 15K bytes - Viewed (0) -
tensorflow/c/experimental/grappler/grappler_test.cc
}; tensorflow::Status status = InitGraphPlugin(plugin_init); ASSERT_EQ(status.code(), tensorflow::error::FAILED_PRECONDITION); ASSERT_EQ(status.message(), "'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();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 13 22:30:58 UTC 2023 - 11.6K bytes - Viewed (0)