struct habana::custom_op::NodeDesc

Overview

Below is a helper structure to represent TPC kernel details:

//! helper struct to represent TPC kernel details
struct NodeDesc {
    //! Unique TPC kernel guid
    std::string tpc_guid;
    //! schema name as used in TORCH_LIBRARY
    std::string schema_name;
    //! TPC kernel params callback function
    allocate_user_params_func user_param_func = nullptr;
};

Fields

• Unique Identifier of TPC kernel:

  std::string guid
  

• Unique Identifier os PyTorch schema:

  std::string schema_name;
  

• Pointer to a callback function which generates parameters data bound to the TPC kernel type:

  allocate_user_params_func user_param_func;