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