Searched defs:cache_miss (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | pmap_lib.cc | 195 PmapFunction(py::function fun, py::function cache_miss, in PmapFunction()
226 const py::function& cache_miss() const { return cache_miss_; } in cache_miss() function in jax::PmapFunction
255 py::function fun, cache_miss, python_shard_arg_fallback; in ClearPythonReferences() local
709 py::function cache_miss, in InitializePmapFunction()
719 py::object MakePmapFunction(py::function fun, py::function cache_miss, in MakePmapFunction()
900 py::function cache_miss = py::cast<py::function>(pickle["cache_miss"]); in BuildPmapSubmodule() local
952 py::function python_shard_arg_fallback) -> py::object { in BuildPmapSubmodule()
|
| D | jax_jit.cc | 506 const py::function& cache_miss() const { return cache_miss_; } in cache_miss() function in jax::__anon838f46370811::CompiledFunction
518 py::function fun, cache_miss, get_device; in ClearPythonReferences() local
605 CompiledFunction::CompiledFunction(py::function fun, py::function cache_miss, in CompiledFunction()
1200 py::function fun, py::function cache_miss, in InitializeCompiledFunction()
1215 py::object MakeCompiledFunction(py::function fun, py::function cache_miss, in MakeCompiledFunction()
1356 py::function cache_miss = py::cast<py::function>(pickle["cache_miss"]); in BuildJaxjitSubmodule() local
1401 std::shared_ptr<CompiledFunctionCache> cache) -> py::object { in BuildJaxjitSubmodule()
|
| /external/pytorch/torch/csrc/inductor/aoti_eager/ |
| D | kernel_holder.cpp | 440 void AOTIPythonKernelHolder::cache_miss( in cache_miss() function in torch::inductor::AOTIPythonKernelHolder
|
| /external/elfutils/debuginfod/ |
| D | debuginfod-client.c | 1135 time_t cache_miss; in debuginfod_query_server() local
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_functional_ops.cc | 1266 bool cache_miss = !partition_cache_.count(cache_hash); in ComputeAsync() local
|