Home
last modified time | relevance | path

Searched refs:GetCompiledMemoryStats (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dpy_executable.h96 StatusOr<CompiledMemoryStats> GetCompiledMemoryStats() const { in GetCompiledMemoryStats() function
97 return executable_->GetCompiledMemoryStats(); in GetCompiledMemoryStats()
H A Dxla.cc376 .def("get_compiled_memory_stats", &PyExecutable::GetCompiledMemoryStats) in PYBIND11_MODULE()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dpjrt_executable.h66 virtual StatusOr<CompiledMemoryStats> GetCompiledMemoryStats() const { in GetCompiledMemoryStats() function