struct habana::custom_op::OutputDesc

Overview

Below is a helper structure to represent custom op output:

//! helper struct for output of op
struct OutputDesc {
    //! output index
    unsigned index;
    //! Tensor output type, default is float
    c10::ScalarType dtype = c10::ScalarType::Float;
    //! output shape calculation callback function
    compute_output_shape_function compute_output_shape_func = nullptr;
};

Fields

• Index of the output to op:

  unsigned index;
  

• Tensor output type, default is float:

  c10::ScalarType dtype = c10::ScalarType::Float;
  

• Output shape calculation callback function:

  compute_output_shape_function compute_output_shape_func = nullptr;