Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dtypes.cc271 StatusOr<PythonBufferTree> GetPythonBufferTree(const py::object& argument) { in GetPythonBufferTree()
272 PythonBufferTree tree; in GetPythonBufferTree()
277 TF_ASSIGN_OR_RETURN(PythonBufferTree subtree, in GetPythonBufferTree()
H A Dtypes.h91 struct PythonBufferTree { struct
98 StatusOr<PythonBufferTree> GetPythonBufferTree( argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
H A Dtpu_client_extension.cc120 TF_ASSIGN_OR_RETURN(PythonBufferTree tree, in PYBIND11_MODULE()