Add possibility to override nvcc arch for tensorflow_jit and compile...

Add possibility to override nvcc arch for tensorflow_jit and compile tensorflow module without loading