Function rematable_storage

oneflow/api/python/remat/remat.cpp:30–35

namespace {
Maybe<vm::RematableTensorStorage> rematable_storage(const std::shared_ptr<one::Tensor>& tensor) {
  auto ret = std::dynamic_pointer_cast<vm::RematableTensorStorage>(
      JUST(tensor->eager_blob_object())->tensor_storage());
  CHECK_NOTNULL_OR_RETURN(ret);
  return ret;
}
}  // namespace

ONEFLOW_API_PYBIND11_MODULE("remat", m) {