Commit 75cee522 by masahi Committed by Tianqi Chen

fix cudnn output shape (#708)

parent 51223c38
......@@ -114,7 +114,7 @@ TVM_REGISTER_GLOBAL("tvm.contrib.cudnn.conv2d.output_shape")
int x_dim3 = args[10];
int w_dim0 = args[11];
int w_dim1 = args[12];
int w_dim2 = args[12];
int w_dim2 = args[13];
int w_dim3 = args[14];
void *out_shape = args[15];
// Set Format
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment