Searched defs:PyTensorInfo (Results 1 – 1 of 1) sorted by relevance
480 struct PyTensorInfo final { struct481 explicit PyTensorInfo( in PyTensorInfo() argument486 py::tuple sizes() const { in sizes()495 int8_t dtype() const { in dtype()500 bool is_memory_planned() const { in is_memory_planned()504 size_t nbytes() const { in nbytes()508 std::string repr() const { in repr()529 std::shared_ptr<Module> module_;530 torch::executor::TensorInfo info_;