|
|
|
@ -174,14 +174,14 @@ OPERATOR_ONNX_CONVERT_DEFINE(Sigmoid, Sigmoid, OpNameInfo())
|
|
|
|
|
OPERATOR_ONNX_CONVERT_DEFINE(Flatten, Flatten, OpNameInfo())
|
|
|
|
|
OPERATOR_ONNX_CONVERT_DEFINE(Squeeze, Squeeze,
|
|
|
|
|
OpNameInfo().Attr("axis", "axes", onnx::AttributeProto_AttributeType_INTS,
|
|
|
|
|
SetAttrTupleValueToProto))
|
|
|
|
|
SetAttrTupleValueToProto<0>))
|
|
|
|
|
|
|
|
|
|
OPERATOR_ONNX_CONVERT_DEFINE(
|
|
|
|
|
Conv2D, Conv,
|
|
|
|
|
OpNameInfo()
|
|
|
|
|
.Attr("dilation", "dilations", onnx::AttributeProto_AttributeType_INTS, SetAttrValueToProto<Int32Imm, 2>)
|
|
|
|
|
.Attr("group", "group", onnx::AttributeProto_AttributeType_INT, SetAttrValueToProto<Int32Imm>)
|
|
|
|
|
.Attr("kernel_size", "kernel_shape", onnx::AttributeProto_AttributeType_INTS, SetAttrTupleValueToProto)
|
|
|
|
|
.Attr("kernel_size", "kernel_shape", onnx::AttributeProto_AttributeType_INTS, SetAttrTupleValueToProto<0>)
|
|
|
|
|
.Attr("pad_mode", "auto_pad", onnx::AttributeProto_AttributeType_STRING,
|
|
|
|
|
[](ValuePtr value, onnx::AttributeProto_AttributeType, onnx::AttributeProto* const attr_proto,
|
|
|
|
|
const PrimitivePtr& prim) {
|
|
|
|
|