yxliu-ntu / check_ops_efficiency

Code for Checking Some Ops Efficiency in PyTorch and Tensorflow

Set up Environment

make clean all
source ./venv/bin/activate

Check on PyTorch

python torch_exp.py

Check on Tensorflow

python tf_exp.py (--eager|--graph)

Run in eager/graph execution mode.