You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by GitBox <gi...@apache.org> on 2020/11/03 22:44:43 UTC
[GitHub] [incubator-tvm] alexgl-github edited a comment on pull request #6840: conv1d_transpose speedup
alexgl-github edited a comment on pull request #6840:
URL: https://github.com/apache/incubator-tvm/pull/6840#issuecomment-721412452
Speedup transposed 1d convolution by eliminating unnecessary multiplications by data values which contain zeroes, for stride greater then 1.
Below are current vs proposed latency numbers for various transposed conv1d parameters:
`
latency current=0.06837 sec new=0.00049 sec channels_out= 1 input_shape=(1, 257, 128) kernel_size=512 strides=128 padding=256
latency current=0.53093 sec new=0.10341 sec channels_out= 257 input_shape=(1, 257, 128) kernel_size=512 strides=4 padding=256
latency current=0.00292 sec new=0.00307 sec channels_out= 1 input_shape=(1, 257, 128) kernel_size=512 strides=1 padding=256
latency current=0.00474 sec new=0.00171 sec channels_out= 1 input_shape=(1, 257, 128) kernel_size=512 strides=2 padding=256
latency current=0.00955 sec new=0.00056 sec channels_out= 1 input_shape=(1, 257, 128) kernel_size=512 strides=16 padding current=256
latency current=0.00054 sec new=0.00023 sec channels_out= 1 input_shape=(1, 1, 16384) kernel_size=512 strides=2 padding=256
latency current=0.00385 sec new=0.00089 sec channels_out= 4 input_shape=(1, 1, 16384) kernel_size=512 strides=4 padding=256
latency current=0.00013 sec new=0.00004 sec channels_out= 1 input_shape=(1, 1, 1024) kernel_size=512 strides=5 padding=256
latency current=0.00002 sec new=0.00002 sec channels_out= 32 input_shape=(1, 3, 224) kernel_size=5 strides=1 padding=0
latency current=0.00004 sec new=0.00003 sec channels_out= 32 input_shape=(1, 3, 224) kernel_size=5 strides=2 padding=0
latency current=0.00006 sec new=0.00003 sec channels_out= 128 input_shape=(1, 32, 32) kernel_size=5 strides=2 padding=0
`
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
users@infra.apache.org