Gather doesn't work with arange and rank 2 index

[Mgluhovskoi]

x_shape, axis, indices = ((3, 3, 2), 1 , (0,2))
x = np.arange(np.prod(x_shape)).reshape(x_shape)
indices_tp = tp.Tensor(indices)
a = tp.Tensor(x)
out = tp.gather(a, axis, indices_tp)
print(out) # throws exception

data = tp.iota((3, 3, 2))
indices = tp.Tensor([0, 2], dtype=tp.int32)
output = tp.gather(data, 1, indices)
print(output) # passes

The following works fine though:

x_shape, axis, indices = ((2, 3), 0, (1))
x = np.arange(np.prod(x_shape)).reshape(x_shape)
indices_tp = tp.Tensor(indices)
a = tp.Tensor(x)
out = tp.gather(a, axis, indices_tp)
print(out)

[Mgluhovskoi]

Issue is related to int64 not yet being supported. Should be fixed along with #116