struct habana::custom_op::InputDesc

Overview

Below is a 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;
};

Fields

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

  input_type type;
  

• Index of the input to op:

  unsigned index;