Searched refs:layout_from_backend (Results 1 – 4 of 4) sorted by relevance
30 return layout_from_backend(backend_); in layout()34 return layout_from_backend(backend()) == kSparse; in is_sparse()38 return layout_from_backend(backend()) == kSparseCsr; in is_sparse_csr()
31 inline Layout layout_from_backend(Backend backend) { in layout_from_backend() function
246 (THPLayout*)Py_NewRef(torch::getTHPLayout(layout_from_backend(backend))); in set_type()
237 …return to(options().device(backendToDeviceType(b)).layout(layout_from_backend(b)), /*non_blocking*… in toBackend()