Searched refs:remove_existing_batch_dim (Results 1 – 2 of 2) sorted by relevance
48 static std::pair<Tensor,int64_t> remove_existing_batch_dim( in remove_existing_batch_dim() function118 auto [self_without_bdim, newly_exposed_logical_dim] = remove_existing_batch_dim(batched, level); in _remove_batch_dim()
95 static std::pair<Tensor, int64_t> remove_existing_batch_dim( in remove_existing_batch_dim() function165 remove_existing_batch_dim(batched, level); in _remove_batch_dim()