Closed robertknight closed 2 days ago
Improve code path for tensors with 5+ dimensions in TensorBase::init_from. Instead of falling back to slow iteration via TensorBase::iter, iterate over inner views of 4 dims and use the faster code path that handles this.
TensorBase::init_from
TensorBase::iter
This was encountered while testing https://huggingface.co/briaai/RMBG-2.0.
Improve code path for tensors with 5+ dimensions in
TensorBase::init_from. Instead of falling back to slow iteration viaTensorBase::iter, iterate over inner views of 4 dims and use the faster code path that handles this.This was encountered while testing https://huggingface.co/briaai/RMBG-2.0.