Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dpy_buffer.cc168 pybind11::tuple PyBuffer::python_shape() const { in python_shape() function in xla::PyBuffer
192 auto shape = py::cast<std::string>(py::str(python_shape())); in CopyToDevice()
243 auto shape = py::cast<std::string>(py::str(python_shape())); in CopyToHostAsync()
H A Dpy_buffer.h135 pybind11::tuple python_shape() const;
H A Djax_jit.cc1441 buffer.buf()->python_shape(), buffer.buf()->python_dtype(), in BuildJaxjitSubmodule()