Searched refs:numpy_array (Results 1 – 4 of 4) sorted by relevance
460 py::array numpy_array = py::cast<py::array>(h); in ArgSignatureOfValue() local461 if (IsFloat0(numpy_array)) { in ArgSignatureOfValue()467 const py::dtype raw_dtype = numpy_array.dtype(); in ArgSignatureOfValue()480 reinterpret_cast<const xla::int64*>(numpy_array.shape()), in ArgSignatureOfValue()481 numpy_array.ndim()), in ArgSignatureOfValue()485 xla::DtypeToPrimitiveType(numpy_array.dtype())); in ArgSignatureOfValue()489 reinterpret_cast<const xla::int64*>(numpy_array.shape()), in ArgSignatureOfValue()490 numpy_array.ndim()), in ArgSignatureOfValue()714 py::array numpy_array = py::cast<py::array>(h); in HandleNdarray() local715 if (IsFloat0(numpy_array)) { in HandleNdarray()[all …]
44 from numpy import ndarray as numpy_array unknown46 numpy_array = None variable70 if numpy_array:636 return numpy_array(buffer=buf, shape=shape, strides=strides,1906 if numpy_array:1934 if numpy_array:1985 if numpy_array:2265 if numpy_array:2266 na = numpy_array(buffer=bytearray(lst),2290 if numpy_array:[all …]
6 from pybind11_tests import numpy_array as m
135 TEST_SUBMODULE(numpy_array, sm) { in TEST_SUBMODULE() argument