add pytorch aten_split_with_sizes
@@ -32,8 +32,8 @@ NN_KERNEL_NAME = {"paddle.nn.BatchNorm": "bn",
"paddle.nn.Softmax": "softmax",
"paddle.nn.Softplus": "softplus",
"paddle.nn.Tanh": "tanh",
- "paddle.nn.AvgPool2D": "pool",
- "paddle.nn.MaxPool2D": "pool",
+ "paddle.nn.AvgPool2D": "avgpool",
+ "paddle.nn.MaxPool2D": "maxpool",
"paddle.nn.Pad1D": "pad",
"paddle.nn.Pad2D": "pad",
"paddle.nn.Pad3D": "pad",
