- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for InitGraphPlugin (0.26 sec)
-
tensorflow/c/experimental/grappler/grappler_test.cc
TF_Status* const status) -> void { TF_SetStatus(status, TF_OK, ""); PopulateDefaultParam(params); params->device_type = "Device1"; }; TF_ASSERT_OK(InitGraphPlugin(plugin_init_0)); TF_ASSERT_OK(InitGraphPlugin(plugin_init_1)); ASSERT_EQ(PluginGraphOptimizerRegistry::CreateOptimizers( std::set<string>{"Device0", "Device1"}) .size(), 2); }
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_internal.h
TF_Status* const); // 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 {
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) -
tensorflow/c/experimental/grappler/grappler.cc
// Step 2: Call `TF_InitPlugin` auto init_fn = reinterpret_cast<TFInitGraphPluginFn>(dso_symbol); return InitGraphPlugin(init_fn); } absl::Status InitGraphPlugin(TFInitGraphPluginFn init_fn) { TP_OptimizerRegistrationParams params{ TP_OPTIMIZER_REGISTRATION_PARAMS_STRUCT_SIZE}; TP_Optimizer optimizer{TP_OPTIMIZER_STRUCT_SIZE}; TP_OptimizerConfigs optimizer_configs{TP_OPTIMIZER_CONFIGS_STRUCT_SIZE};
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Sep 06 19:12:29 GMT 2023 - 15K bytes - Viewed (1)