| 963 | } |
| 964 | |
| 965 | std::optional<InnerJVPFunction> make_jvp_function( |
| 966 | nb::object input_structure) { |
| 967 | if (!jvp_fun_.has_value()) { |
| 968 | return std::nullopt; |
| 969 | } |
| 970 | |
| 971 | return InnerJVPFunction(*jvp_fun_, input_structure); |
| 972 | } |
| 973 | |
| 974 | std::optional<InnerVmapFunction> make_vmap_function( |
| 975 | nb::object input_structure) { |
nothing calls this directly
no test coverage detected