|
|
|
@@ -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) { |
|
|
|
|