Created by: chengduoZH
Add dtype in coalesce_tensor_op, because the output data type may be different from the data type of the input tensors.