• Arjun Jain's avatar
    - fixed a bug in cafe conv (values of _M, _N, _K) · c649d668
    Arjun Jain 提交于
    - added a test that does a check on variety of shapes and sizes of image and kernel
    - removed flip form local_conv_gemm in cuda/opt.py is the code never reached there for me
    - added flip in the kernel in the test code itself
    c649d668
test_conv_cuda_ndarray.py 38.1 KB