Searched defs:hx_opt (Results 1 – 1 of 1) sorted by relevance
611 torch::optional<std::tuple<Tensor, Tensor>> hx_opt) { in forward_helper() argument668 torch::optional<std::tuple<Tensor, Tensor>> hx_opt) { in forward() argument684 torch::optional<std::tuple<Tensor, Tensor>> hx_opt) { in forward_with_packed_input() argument949 torch::optional<std::tuple<Tensor, Tensor>> hx_opt) { in forward() argument