Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Ddlpack.h28 StatusOr<pybind11::capsule> BufferToDLPackManagedTensor(pybind11::handle buffer,
H A Ddlpack.cc266 StatusOr<py::capsule> BufferToDLPackManagedTensor(py::handle py_buffer, in BufferToDLPackManagedTensor() function
H A Dxla.cc406 m.def("buffer_to_dlpack_managed_tensor", BufferToDLPackManagedTensor, in PYBIND11_MODULE()