Searched defs:collapseDims (Results 1 – 2 of 2) sorted by relevance
68 TensorInfo<T, IndexType>::collapseDims(const int excludeDim) { in collapseDims() function in at::cuda::detail::TensorInfo
70 void collapseDims() {}; in collapseDims() function72 void collapseDims(TensorInfo<T, T2>& info, Args&... infos) { in collapseDims() function