Skip to content

Conversation

percevalw
Copy link
Member

Description

  • Add map_indices and make_indices_ranges with C++ backends and expose lengths.map_indices and lengths.make_indices_ranges with contiguous indice_dims, boundary handling, and flat indices with offsets and span ids for pooling with embedding_bag
  • Introduce FoldedTensorLayout to store full_names and data_dims with named dimension resolution and helper methods and use it as the lengths container for FoldedTensor
  • Improve as_folded_tensor to better infer dims and dtype from nested data and to accept named data_dims and better handle names and empty structures
  • Benchmark script adds --cases to run selected cases and a new case for range based pooling and adjusts outputs

Checklist

  • If this PR is a bug fix, the bug is documented in the test suite.
  • Changes were documented in the changelog (pending section).
  • If necessary, changes were made to the documentation.

Copy link

github-actions bot commented Sep 10, 2025

Coverage Report

NameStmtsMiss∆ MissCover
TOTAL27600100.00%

1 file skipped due to complete coverage.

Coverage success: total of 100.00% is above 100.00% 🎉

@percevalw percevalw force-pushed the indice-mapping branch 3 times, most recently from b019d9a to f6e9bad Compare September 10, 2025 17:44
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant