| 636 | } |
| 637 | |
| 638 | void TensorWrapper::set_module_trace_info(PyObject* obj) { |
| 639 | // TODO: erase when obj == nullptr |
| 640 | ModuleTraceTransformation::module_trace_info_map[m_tensor->data()] = |
| 641 | py::reinterpret_borrow<py::object>(obj); |
| 642 | } |
| 643 | |
| 644 | void TensorWrapper::_set_format(PyObject* dest) { |
| 645 | auto py_dest = py::reinterpret_borrow<py::object>(dest); |