Searched defs:jagged (Results 1 – 3 of 3) sorted by relevance
371 def buffer_from_jagged(jagged): argument
193 auto jagged = nt_impl->get_buffer().view({-1, nt.size(2), nt.size(3)}); in matmul_nested_with_broadcasted_dense() local
1042 auto jagged = nt_impl->get_buffer().view(jagged_size); in cat_nested_as_jagged() local