Add tensor parallelism to the paged llama model

Add tensor parallelism to the paged llama model #1021