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

Method reset

python/src/transforms.cpp:939–950  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

937 return *this;
938 }
939 void reset() {
940 fun_.reset();
941 if (vjp_fun_.has_value()) {
942 (*vjp_fun_).reset();
943 }
944 if (jvp_fun_.has_value()) {
945 (*jvp_fun_).reset();
946 }
947 if (vmap_fun_.has_value()) {
948 (*vmap_fun_).reset();
949 }
950 }
951
952 friend int py_custom_function_tp_traverse(PyObject*, visitproc, void*);
953

Callers 8

~PyCompiledFunMethod · 0.45
~PyCheckpointedFunMethod · 0.45
~InnerFunctionMethod · 0.45
~InnerFunctionMethod · 0.45
~InnerVJPFunctionMethod · 0.45
~InnerJVPFunctionMethod · 0.45
~InnerVmapFunctionMethod · 0.45

Calls

no outgoing calls

Tested by

no test coverage detected