Operator Registration APIs

Operator type registration starts with the REG_OP API and ends with the OP_END API. The input, output, and attribute information (INPUT, OUTPUT, and ATTR) to be registered is linked by periods (.). After an operator type is registered, a class named after the operator type is automatically generated.

Instance

REG_OP(FullConnection)

.INPUT(x, TensorType::ALL())

.INPUT(w, TensorType::ALL())

.INPUT(b, TensorType::ALL())

.OUTPUT(y, TensorType::ALL())

.ATTR(num_output, AttrValue::INT{0})

.INFER_SHAPE_AND_TYPE(FullConnectionInfer)

.ATTR_ALL_VERIFY(FullConnectionVerify)

.OP_END()

The operator registration APIs are defined in operator_reg.h. For details about registered operators and their header files, see List of Built-in Operators.

• REG_OP
  
• ATTR
  
• REQUIRED_ATTR
  
• INPUT
  
• OPTIONAL_INPUT
  
• DYNAMIC_INPUT
  
• OUTPUT
  
• DYNAMIC_OUTPUT
  
• INFER_SHAPE_AND_TYPE
  
• ATTR_ALL_VERIFY
  
• OP_END
  
• List of Built-in Operators