struct habana::custom_op::NodeDesc


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;

Detailed Documentation


  • 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;