Searched refs:python_dtype (Results 1 – 3 of 3) sorted by relevance
136 pybind11::dtype python_dtype() const;
172 pybind11::dtype PyBuffer::python_dtype() const { in python_dtype() function in xla::PyBuffer193 auto dtype = py::cast<std::string>(py::str(python_dtype())); in CopyToDevice()244 auto dtype = py::cast<std::string>(py::str(python_dtype())); in CopyToHostAsync()
1441 buffer.buf()->python_shape(), buffer.buf()->python_dtype(), in BuildJaxjitSubmodule()