Searched defs:elems_flat (Results 1 – 2 of 2) sorted by relevance
556 def _elems_flat_to_batchable(elems_flat): argument
185 def generic_associative_scan(operator, elems_flat, dim=0): argument