curobo.curobolib.lbfgs_step_cu module

curobo.curobolib.lbfgs_step_cu.debug_reduce(arg0: torch.Tensor, arg1: torch.Tensor, arg2: torch.Tensor, arg3: torch.Tensor, arg4: int, arg5: int, arg6: int) List[torch.Tensor]

L-BFGS Debug

curobo.curobolib.lbfgs_step_cu.forward(arg0: torch.Tensor, arg1: torch.Tensor, arg2: torch.Tensor, arg3: torch.Tensor, arg4: torch.Tensor, arg5: torch.Tensor, arg6: torch.Tensor, arg7: torch.Tensor, arg8: float, arg9: int, arg10: int, arg11: int, arg12: bool) List[torch.Tensor]

L-BFGS Update + Step (CUDA)

curobo.curobolib.lbfgs_step_cu.step(arg0: torch.Tensor, arg1: torch.Tensor, arg2: torch.Tensor, arg3: torch.Tensor, arg4: torch.Tensor, arg5: float, arg6: int, arg7: int, arg8: int) List[torch.Tensor]

L-BFGS step (CUDA)

curobo.curobolib.lbfgs_step_cu.update(arg0: torch.Tensor, arg1: torch.Tensor, arg2: torch.Tensor, arg3: torch.Tensor, arg4: torch.Tensor, arg5: torch.Tensor, arg6: torch.Tensor, arg7: int, arg8: int, arg9: int) List[torch.Tensor]

L-BFGS Update (CUDA)