Skip to content
Snippets Groups Projects
Commit 75cee522 authored by masahi's avatar masahi Committed by Tianqi Chen
Browse files

fix cudnn output shape (#708)

parent 51223c38
No related branches found
No related tags found
No related merge requests found
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment