Searched refs:fun_ (Results 1 – 4 of 4) sorted by relevance
2 name_regexp = ^fun_
477 : fun_(ActionAdapter{::std::shared_ptr<ActionInterface<F>>(impl)}) {}483 explicit Action(const Action<Func>& action) : fun_(action.fun_) {}486 bool IsDoDefault() const { return fun_ == nullptr; }498 return internal::Apply(fun_, ::std::move(args));507 fun_ = ::std::forward<G>(g);512 fun_ = IgnoreArgs<typename ::std::decay<G>::type>{::std::forward<G>(g)};526 ::std::function<F> fun_;
136 : fun_(std::move(fun)), in PmapFunction()159 return inspect->attr("signature")(fun_); in PythonSignature()174 const py::function fun_; // The Python function to pmap. member in jax::PmapFunction
860 return inspect->attr("signature")(fun_); in PythonSignature()876 const py::function fun_; // The Python function to jit. member in jax::__anone04173081411::CompiledFunction921 : fun_(std::move(fun)), in CompiledFunction()1143 return fun_(*args, **kwargs); in Call()