Change tensor utils names (#7284)

## Summary

It doesn't really make a lot of sense to call these `extension_` utils
since really it is specific to tensor types (fixed shape tensor,
vectors, etc).

Am open to calling these `fixed_tensor_` instead as well.

## Testing

N/A

Signed-off-by: Connor Tsui <connor.tsui20@gmail.com>

Author: connortsui20

vortex-tensor/src/scalar_fns/cosine_similarity.rs

@@ -34,7 +34,7 @@ use crate::matcher::AnyTensor;
 use crate::scalar_fns::ApproxOptions;
 use crate::scalar_fns::inner_product::InnerProduct;
 use crate::scalar_fns::l2_norm::L2Norm;
-use crate::utils::extension_element_ptype;
+use crate::utils::tensor_element_ptype;
 
 /// Cosine similarity between two columns.
 ///
@@ -128,7 +128,7 @@ impl ScalarFnVTable for CosineSimilarity {
             "CosineSimilarity inputs must be an `AnyTensor`, got {lhs}"
         );
 
-        let ptype = extension_element_ptype(lhs_ext)?;
+        let ptype = tensor_element_ptype(lhs_ext)?;
         vortex_ensure!(
             ptype.is_float(),
             "CosineSimilarity element dtype must be a float primitive, got {ptype}"

vortex-tensor/src/scalar_fns/inner_product.rs

@@ -33,9 +33,9 @@ use crate::encodings::turboquant::TurboQuant;
 use crate::encodings::turboquant::compute::cosine_similarity;
 use crate::matcher::AnyTensor;
 use crate::scalar_fns::ApproxOptions;
-use crate::utils::extension_element_ptype;
-use crate::utils::extension_list_size;
 use crate::utils::extract_flat_elements;
+use crate::utils::tensor_element_ptype;
+use crate::utils::tensor_list_size;
 
 /// Inner product (dot product) between two columns.
 ///
@@ -127,7 +127,7 @@ impl ScalarFnVTable for InnerProduct {
             "InnerProduct inputs must be an `AnyTensor`, got {lhs}"
         );
 
-        let ptype = extension_element_ptype(lhs_ext)?;
+        let ptype = tensor_element_ptype(lhs_ext)?;
         vortex_ensure!(
             ptype.is_float(),
             "InnerProduct element dtype must be a float primitive, got {ptype}"
@@ -168,7 +168,7 @@ impl ScalarFnVTable for InnerProduct {
         // Get list size from the dtype. Both sides have the same dtype (validated by
         // `return_dtype`).
         let ext = lhs.dtype().as_extension();
-        let list_size = extension_list_size(ext)? as usize;
+        let list_size = tensor_list_size(ext)? as usize;
 
         // Extract the storage array from each extension input. We pass the storage (FSL) rather
         // than the extension array to avoid canonicalizing the extension wrapper.

vortex-tensor/src/scalar_fns/l2_norm.rs

@@ -32,9 +32,9 @@ use vortex_error::vortex_err;
 use crate::encodings::turboquant::TurboQuant;
 use crate::matcher::AnyTensor;
 use crate::scalar_fns::ApproxOptions;
-use crate::utils::extension_element_ptype;
-use crate::utils::extension_list_size;
 use crate::utils::extract_flat_elements;
+use crate::utils::tensor_element_ptype;
+use crate::utils::tensor_list_size;
 
 /// L2 norm (Euclidean norm) of a tensor or vector column.
 ///
@@ -109,7 +109,7 @@ impl ScalarFnVTable for L2Norm {
             "L2Norm input must be an `AnyTensor`, got {input_dtype}"
         );
 
-        let ptype = extension_element_ptype(ext)?;
+        let ptype = tensor_element_ptype(ext)?;
         vortex_ensure!(
             ptype.is_float(),
             "L2Norm element dtype must be a float primitive, got {ptype}"
@@ -144,7 +144,7 @@ impl ScalarFnVTable for L2Norm {
 
         // Get element ptype and list size from the dtype (validated by `return_dtype`).
         let ext = input.dtype().as_extension();
-        let list_size = extension_list_size(ext)? as usize;
+        let list_size = tensor_list_size(ext)? as usize;
 
         let storage = input.data().storage_array();
         let flat = extract_flat_elements(storage, list_size, ctx)?;

vortex-tensor/src/utils.rs

@@ -20,7 +20,7 @@ use vortex_error::vortex_err;
 /// Extracts the list size from a tensor-like extension dtype.
 ///
 /// The storage dtype must be a `FixedSizeList`.
-pub fn extension_list_size(ext: &ExtDTypeRef) -> VortexResult<u32> {
+pub fn tensor_list_size(ext: &ExtDTypeRef) -> VortexResult<u32> {
     let DType::FixedSizeList(_, list_size, _) = ext.storage_dtype() else {
         vortex_bail!(
             "expected FixedSizeList storage dtype, got {}",
@@ -34,7 +34,7 @@ pub fn extension_list_size(ext: &ExtDTypeRef) -> VortexResult<u32> {
 /// Extracts the float element [`PType`] from a tensor-like extension dtype.
 ///
 /// The storage dtype must be a `FixedSizeList` of non-nullable primitives.
-pub fn extension_element_ptype(ext: &ExtDTypeRef) -> VortexResult<PType> {
+pub fn tensor_element_ptype(ext: &ExtDTypeRef) -> VortexResult<PType> {
     let element_dtype = ext
         .storage_dtype()
         .as_fixed_size_list_element_opt()