Import as_tensor_variable

Make sure to import as_tensor_variable