Searched refs:PyQnnTensorWrapper (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/executorch/backends/qualcomm/aot/python/ |
| H A D | PyQnnWrapperAdaptor.cpp | 107 PYBIND11_NUMPY_DTYPE(PyQnnTensorWrapper::EncodingData, scale, offset); in PYBIND11_MODULE()
240 py::class_<PyQnnTensorWrapper::Encoding>(m, "Encoding") in PYBIND11_MODULE()
241 .def_readonly("data", &PyQnnTensorWrapper::Encoding::data) in PYBIND11_MODULE()
242 .def_readonly("axis", &PyQnnTensorWrapper::Encoding::axis); in PYBIND11_MODULE()
244 py::class_<PyQnnTensorWrapper, std::shared_ptr<PyQnnTensorWrapper>>( in PYBIND11_MODULE()
247 .def("GetDims", &PyQnnTensorWrapper::GetDims) in PYBIND11_MODULE()
248 .def("GetDataType", &PyQnnTensorWrapper::GetDataType) in PYBIND11_MODULE()
249 .def("GetName", &PyQnnTensorWrapper::GetName) in PYBIND11_MODULE()
250 .def("GetEncodings", &PyQnnTensorWrapper::GetEncodings); in PYBIND11_MODULE()
|
| H A D | PyQnnWrapperAdaptor.h | 101 class PyQnnTensorWrapper {
103 explicit PyQnnTensorWrapper(const std::shared_ptr<TensorWrapper>& wrapper) { in PyQnnTensorWrapper() function
|
| H A D | PyQnnManagerAdaptor.h | 273 ret.append(PyQnnTensorWrapper(input)); in GetGraphInputs()
282 ret.append(PyQnnTensorWrapper(output)); in GetGraphOutputs()
|