struct habana::custom_op::OutputDesc

Overview

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

Detailed Documentation

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;