struct habana::custom_op::NodeDesc

Overview

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

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;