ChannelPrunedLearner with tf.keras.layers.Conv1D

[Augnine]

I use tf.keras.layers.Conv1D in my own network, and i have trained it with FullPrecLearner. Then when I use ChannelPrunedLearner, there are some errors :

ERROR:tensorflow:the number of real inputs of model/encoder/encode_conv1d/conv1d/Conv2D should be 1
ERROR:tensorflow:the number of weights variable of model/encoder/encode_conv1d/conv1d/Conv2D should be 1

the structure of con1d like this:
name: "model/encoder/encode_conv1d/conv1d/Conv2D"
op: "Conv2D"
input: "model/encoder/encode_conv1d/conv1d/ExpandDims"
input: "model/encoder/encode_conv1d/conv1d/ExpandDims_1"
attr {
key: "T"
value {
type: DT_FLOAT
}
}
attr {
key: "data_format"
value {
s: "NHWC"
}
}
attr {
key: "dilations"
value {
list {
i: 1
i: 1
i: 1
i: 1
}
}
}
attr {
key: "padding"
value {
s: "VALID"
}
}
attr {
key: "strides"
value {
list {
i: 1
i: 1
i: 10
i: 1
}
}
}
attr {
key: "use_cudnn_on_gpu"
value {
b: true
}
}