Searched refs:CompiledFunction (Results 1 – 2 of 2) sorted by relevance
841 class CompiledFunction { class843 CompiledFunction(py::function fun, py::function cache_miss,847 ~CompiledFunction();916 CompiledFunction::CompiledFunction(py::function fun, py::function cache_miss, in CompiledFunction() function in jax::__anone04173081411::CompiledFunction930 CompiledFunction::~CompiledFunction() { in ~CompiledFunction()1026 CacheEntry* CompiledFunction::GetCacheEntryIfPresent( in GetCacheEntryIfPresent()1043 CacheEntry* CompiledFunction::AddCacheEntry(const py::args& args, in AddCacheEntry()1105 py::object CompiledFunction::Call(py::args args, py::kwargs kwargs) { in Call()1212 py::class_<CompiledFunction, std::unique_ptr<CompiledFunction>> cfun( in BuildJaxjitSubmodule()1214 cfun.def("__call__", &CompiledFunction::Call); in BuildJaxjitSubmodule()[all …]
525 def CompiledFunction(x): function530 y = CompiledFunction(x)589 def CompiledFunction(x): function594 y = CompiledFunction(x)619 def CompiledFunction(x): function624 y = CompiledFunction(x)