Searched refs:_chunk_cat_out (Results 1 – 2 of 2) sorted by relevance
836 at::native::_chunk_cat_out(tensors, dim, num_chunks, out); in _chunk_cat_out_cuda()
2752 Tensor& _chunk_cat_out(TensorList tensors, int64_t dim, int64_t num_chunks, Tensor& out) { in _chunk_cat_out() function