docstring update
get_theano_vars => get_theano_graph
get_sym_arg => get_symbolic
get_theano_graph doesn't call trace