MCPcopy Create free account
hub / github.com/ml-explore/mlx / make_vjp_function

Method make_vjp_function

python/src/transforms.cpp:955–963  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

953
954 private:
955 std::optional<InnerVJPFunction> make_vjp_function(
956 nb::object input_structure,
957 std::shared_ptr<nb::object> output_structure) {
958 if (!vjp_fun_.has_value()) {
959 return std::nullopt;
960 }
961
962 return InnerVJPFunction(*vjp_fun_, input_structure, output_structure);
963 }
964
965 std::optional<InnerJVPFunction> make_jvp_function(
966 nb::object input_structure) {

Callers

nothing calls this directly

Calls 1

InnerVJPFunctionClass · 0.85

Tested by

no test coverage detected