Functionforeach_instance(metadata, output_dir, func, max_workers=None, desc='Processing objects', log_interval=500, timeout=None)
data_toolkit/datasets/ObjaverseXL.py:83
Functionforeach_instance(metadata, output_dir, func, max_workers=None, desc='Processing objects', timeout=None)
data_toolkit/datasets/TexVerse.py:34
Methodforward(self, x: torch.Tensor, context: Union[Dict[str, torch.Tensor], Tuple[torch.Tensor, torch.Tensor]])
pixal3d/modules/attention/proj_attention.py:38
Methodforward(
self,
x: SparseTensor,
context: Union[Dict[str, Union[torch.Tensor, VarLenTensor,
pixal3d/modules/sparse/attention/proj_attention.py:26
Methodforward(
self,
x: SparseTensor,
context: Union[Dict[str, Union[torch.Tensor, VarLenTensor, Sp
pixal3d/modules/sparse/attention/proj_attention.py:79
Methodforward Args: q (SparseTensor): [..., N, H, D] tensor of queries k (SparseTensor): [..., N, H, D] tensor of keys
pixal3d/modules/sparse/attention/rope.py:35