Searched refs:PyTensorInfo (Results 1 – 1 of 1) sorted by relevance
480 struct PyTensorInfo final { struct481 explicit PyTensorInfo( in PyTensorInfo() function548 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 …]