Make lookup_table_op and softmax_op supporting high rank tensor