struct habana::custom_op::InputDesc

Overview

helper structure to represent Custom op input.

//! helper struct for input to op
struct InputDesc {
    //! input type, e.g: Tensor, Scalar..
    input_type type;
    //! input index
    unsigned index;
};

Detailed Documentation

Fields

• Input to custom op type: Tensor, Scalar, User Param:

input_type type;

• Index of the input to Op:

unsigned index;