Operator注册类

注册算子类型以REG_OP为起始，以“.”链接INPUT、OUTPUT、ATTR等接口注册算子的输入、输出和属性信息，最终以OP_END接口结束。注册算子类型成功后，自动生成以算子类型名称命名的类。

例如：

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()

Operator注册类接口在operator_reg.h中定义。已注册的算子及对应的头文件，请参见内置的算子类型列表。

• REG_OP
  
• ATTR
  
• REQUIRED_ATTR
  
• INPUT
  
• OPTIONAL_INPUT
  
• DYNAMIC_INPUT
  
• OUTPUT
  
• DYNAMIC_OUTPUT
  
• INFER_SHAPE_AND_TYPE
  
• ATTR_ALL_VERIFY
  
• OP_END
  
• 内置的算子类型列表