Searched refs:get_strided_layout_nested_tensor (Results 1 – 1 of 1) sorted by relevance
750 def get_strided_layout_nested_tensor(jagged_layout_nt): function758 query = get_strided_layout_nested_tensor(query)759 key = get_strided_layout_nested_tensor(key)760 value = get_strided_layout_nested_tensor(value)