Searched refs:python_shape (Results 1 – 3 of 3) sorted by relevance
101 pybind11::tuple python_shape() const;
58 pybind11::tuple PyBuffer::python_shape() const { in python_shape() function in xla::PyBuffer
1252 buffer->python_shape(), buffer->python_dtype(), results->weak_type)); in BuildJaxjitSubmodule()