Home
last modified time | relevance | path

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

/aosp_15_r20/external/executorch/extension/pybindings/
H A Dpybindings.cpp480 struct PyTensorInfo final { struct
481 explicit PyTensorInfo( in PyTensorInfo() argument
548 std::unique_ptr<PyTensorInfo> input_tensor_meta(size_t index) const { in input_tensor_meta()
552 return std::make_unique<PyTensorInfo>(module_, result.get()); in input_tensor_meta()
559 std::unique_ptr<PyTensorInfo> output_tensor_meta(size_t index) const { in output_tensor_meta()
563 return std::make_unique<PyTensorInfo>(module_, result.get()); in output_tensor_meta()
1093 py::class_<PyTensorInfo>(m, "TensorInfo") in PYBIND11_MODULE()
1094 .def("sizes", &PyTensorInfo::sizes, call_guard) in PYBIND11_MODULE()
1095 .def("dtype", &PyTensorInfo::dtype, call_guard) in PYBIND11_MODULE()
1096 .def("is_memory_planned", &PyTensorInfo::is_memory_planned, call_guard) in PYBIND11_MODULE()
[all …]