Add primitive test case for varray
Move softmax to array operations