Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for InitGraphPlugin (0.26 sec)

  1. 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)
  2. 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)
  3. 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)
Back to top