Files
pytorch/binaries/bench_gen
Jongsoo Park f1f521f71b make bench_gen.py work for 3d conv (#12433)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/12433

To test 3d conv, we need to pass lists in spec argument. We also don't want to set use_cudnn=True which is the default in brew.

Reviewed By: llyfacebook, csummersea

Differential Revision: D10234315

fbshipit-source-id: 96a39992a97e020d6e9dac103e6d64df0cc1020b
2018-10-08 12:24:43 -07:00
..