Searched refs:AddCacheEntry (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | pmap_lib.cc | 168 PmapCacheEntry* AddCacheEntry(const py::args& args, const py::kwargs& kwargs,
209 PmapCacheEntry* PmapFunction::AddCacheEntry(const py::args& args, in AddCacheEntry() function in jax::PmapFunction
285 cache_entry = AddCacheEntry(args, kwargs, arguments.signature, in Call()
|
| D | jax_jit.cc | 869 CacheEntry* AddCacheEntry(const py::args& args, const py::kwargs& kwargs,
1043 CacheEntry* CompiledFunction::AddCacheEntry(const py::args& args, in AddCacheEntry() function in jax::__anone04173081411::CompiledFunction
1164 cache_entry = AddCacheEntry(args, kwargs, arguments.signature, in Call()
|